school_api.check_code.predict.CheckCode.denoise_img() - Code Metrics - Inspection of "优化验证码识别" - dairoot/school-api - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( 09487a...5fd33b )

by dai

created 2018-09-21 11:53 UTC

school_api.check_code.predict.CheckCode.denoise_img() A

↳ Parent: Project

Complexity

Conditions

Size

Total Lines	9
Code Lines	8

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	4
eloc	8
nop	2
dl	0
loc	9
rs	10
c	0
b	0
f	0

# -*- coding: utf-8 -*-
"""
    school_api.check_code.predict
    ~~~~~~~~~~~~~~~~
    This module provides functions for Identification verification code

    :copyright: (c) 2018 by dairoot.
    :license: MIT, see LICENSE for more details.
"""
import os
from io import BytesIO
import numpy as np
from PIL import Image


class CheckCode(object):

    """ 正方系统验证码识别 """
    data_file = os.path.dirname(os.path.realpath(__file__)) + '/theta.dat'
    real_all_theta = np.matrix(np.loadtxt(data_file)).transpose()

    def __init__(self):
        self.img = None

    def photo_to_text(self):
        '''
        图片转数据
        '''
        x_size, y_size = self.img.size
        y_size -= 5
        piece = (x_size - 22) // 8
        centers = [4 + piece * (2 * i + 1) for i in range(4)]
        photo_data = []
        for center in centers:
            single_img = self.img.crop((center - (piece + 2), 1, center + (piece + 2), y_size))
            width, height = single_img.size
            photo_data_x = []
            for h_index in range(0, height):
                for w_index in range(0, width):
                    pixel = single_img.getpixel((w_index, h_index))
                    photo_data_x.append(int(pixel == 255))
            photo_data.append(photo_data_x)
        return photo_data

    def verify(self, img_stream):
        '''
        将图片转成numpy数组数据 与 训练好的模型 进行匹配
        '''
        obj = BytesIO(img_stream)
        img = Image.open(obj).convert("L")
        self.img = self.denoise_img(img)
        data = np.matrix(self.photo_to_text())
        data = np.hstack((np.ones((data.shape[0], 1)), data))
        all_predict = 1.0 / (1.0 + np.exp(-(np.dot(data, self.real_all_theta))))
        pred = np.argmax(all_predict, axis=1)
        answers = map(chr, map(lambda x: x + 48 if x <= 9 else x + 87, pred))
        return ''.join(answers)

    def denoise_img(self, img):
class Foo:
    def some_method(self, x, y):
        return x + y;
        '''图片降噪处理'''
        img2 = Image.new("L", img.size, 255)
        for x in range(img.size[1]):

            for y in range(img.size[0]):

                pix = img.getpixel((y, x))
                if pix == 17:  # these are the numbers to get
                    img2.putpixel((y, x), 0)
        return img2


1			# -- coding: utf-8 --
2			"""
3			school_api.check_code.predict
4			~~~~~~~~~~~~~~~~
5			This module provides functions for Identification verification code
6
7			:copyright: (c) 2018 by dairoot.
8			:license: MIT, see LICENSE for more details.
9			"""
10			import os
11			from io import BytesIO
12			import numpy as np
13			from PIL import Image
14
15
16			class CheckCode(object):
			0 ignored issues – show Unused Code introduced 2018-07-19 02:41 UTC by Report Bug Copy Issue Report The variable `__class__` seems to be unused. Loading history...
17			""" 正方系统验证码识别 """
18			data_file = os.path.dirname(os.path.realpath(__file__)) + '/theta.dat'
19			real_all_theta = np.matrix(np.loadtxt(data_file)).transpose()
20
21			def __init__(self):
22			self.img = None
23
24			def photo_to_text(self):
25			'''
26			图片转数据
27			'''
28			x_size, y_size = self.img.size
29			y_size -= 5
30			piece = (x_size - 22) // 8
31			centers = [4 + piece * (2 * i + 1) for i in range(4)]
32			photo_data = []
33			for center in centers:
34			single_img = self.img.crop((center - (piece + 2), 1, center + (piece + 2), y_size))
35			width, height = single_img.size
36			photo_data_x = []
37			for h_index in range(0, height):
38			for w_index in range(0, width):
39			pixel = single_img.getpixel((w_index, h_index))
40			photo_data_x.append(int(pixel == 255))
41			photo_data.append(photo_data_x)
42			return photo_data
43
44			def verify(self, img_stream):
45			'''
46			将图片转成numpy数组数据与训练好的模型进行匹配
47			'''
48			obj = BytesIO(img_stream)
49			img = Image.open(obj).convert("L")
50			self.img = self.denoise_img(img)
51			data = np.matrix(self.photo_to_text())
52			data = np.hstack((np.ones((data.shape[0], 1)), data))
53			all_predict = 1.0 / (1.0 + np.exp(-(np.dot(data, self.real_all_theta))))
54			pred = np.argmax(all_predict, axis=1)
55			answers = map(chr, map(lambda x: x + 48 if x <= 9 else x + 87, pred))
56			return ''.join(answers)
57
58			def denoise_img(self, img):
			0 ignored issues – show Coding Style introduced 2018-09-21 11:55 UTC by Report Bug Copy Issue Report This method could be written as a function/class method. If a method does not access any attributes of the class, it could also be implemented as a function or static method. This can help improve readability. For example class Foo: def some_method(self, x, y): return x + y; could be written as class Foo: @classmethod def some_method(cls, x, y): return x + y; Loading history...
59			'''图片降噪处理'''
60			img2 = Image.new("L", img.size, 255)
61			for x in range(img.size[1]):
			0 ignored issues – show Coding Style Naming introduced 2018-09-21 11:55 UTC by Report Bug Copy Issue Report The name `x` does not conform to the variable naming conventions (`(([a-z][a-z0-9_]{2,30})\|(_[a-z0-9_]*))$`). This check looks for invalid names for a range of different identifiers. You can set regular expressions to which the identifiers must conform if the defaults do not match your requirements. If your project includes a Pylint configuration file, the settings contained in that file take precedence. To find out more about Pylint, please refer to their site. Loading history...
62			for y in range(img.size[0]):
			0 ignored issues – show Coding Style Naming introduced 2018-09-21 11:55 UTC by Report Bug Copy Issue Report The name `y` does not conform to the variable naming conventions (`(([a-z][a-z0-9_]{2,30})\|(_[a-z0-9_]*))$`). This check looks for invalid names for a range of different identifiers. You can set regular expressions to which the identifiers must conform if the defaults do not match your requirements. If your project includes a Pylint configuration file, the settings contained in that file take precedence. To find out more about Pylint, please refer to their site. Loading history...
63			pix = img.getpixel((y, x))
64			if pix == 17: # these are the numbers to get
65			img2.putpixel((y, x), 0)
66			return img2
67

dairoot / school-api

Push — master ( 09487a...5fd33b )

school_api.check_code.predict.CheckCode.denoise_img() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like