| | |
|---|
| | | digits.append([cv2.imread(path + "/@-2.jpg",cv2.IMREAD_GRAYSCALE),'@']) |
|---|
| | | digits.append([cv2.imread(path + "/0-0.jpg",cv2.IMREAD_GRAYSCALE),0]) |
|---|
| | | digits.append([cv2.imread(path + "/0-1.jpg",cv2.IMREAD_GRAYSCALE),0]) |
|---|
| | | digits.append([cv2.imread(path + "/0-2.jpg",cv2.IMREAD_GRAYSCALE),0]) |
|---|
| | | digits.append([cv2.imread(path + "/0-3.jpg",cv2.IMREAD_GRAYSCALE),0]) |
|---|
| | | digits.append([cv2.imread(path + "/1-0.jpg",cv2.IMREAD_GRAYSCALE),1]) |
|---|
| | | digits.append([cv2.imread(path + "/1-1.jpg",cv2.IMREAD_GRAYSCALE),1]) |
|---|
| | | digits.append([cv2.imread(path + "/2-0.jpg",cv2.IMREAD_GRAYSCALE),2]) |
|---|
| | |
|---|
| | | for file in filename: |
|---|
| | | image = cv2.imread(input_dir + "/" + file,0) # 读取图片 |
|---|
| | | ret, binary = cv2.threshold(image,0,255,cv2.THRESH_BINARY | cv2.THRESH_OTSU) |
|---|
| | | #ret, binary = cv2.threshold(image, 0, 255, cv2.THRESH_BINARY_INV | cv2.THRESH_OTSU)[1]) |
|---|
| | | num = get_most_simmilar_digit(binary) |
|---|
| | | cv2.imwrite("images52_infer/" + str(num)+"_" + file,binary) # 读取图片 |
|---|
| | | |
|---|
