问题描述
我正在尝试使用不同的字体为OCR生成数据集,但是在一定的for循环中,该迭代给了我并且错误Typeerror: int object is not iterable.
我已经进行了足够的搜索,得出结论,关于StackOverFlow的大多数答案都建议使用我的for循环中的范围,包括(len),但我不确定是否遵循该范围。
功能如下:
def gen_rand_string_data(data_count,min_char_count=3,max_char_count=8,max_char=16,x_pos='side',img_size=(32,256,1),font=cv2.FONT_HERShey_SIMPLEX,font_scale=np.arange(0.7,1,0.1),thickness=range(1,3,1)):
'''
random string data generation
'''
start_time = dt.datetime.Now()
images = []
labels = []
color = (255,255,255)
count = 0
char_list = list(string.ascii_letters) \
+ list(string.digits) \
+ list(' ')
while (1):
for fs in font_scale:
for thick in thickness:
for f in font:
img = np.zeros(img_size,np.uint8)
char_count = np.random.randint(min_char_count,\
(max_char_count + 1))
rand_str = ''.join(np.random.choice(char_list,\
char_count))
# generate image data
text_size = cv2.getTextSize(rand_str,f,fs,thick)[0]
if x_pos == 'side':
org_x = 0
else:
org_x = (img_size[1] - text_size[0]) // 2
org_y = (img_size[0] + text_size[1]) // 2
cv2.putText(img,rand_str,(org_x,org_y),\
color,thick,cv2.LINE_AA)
label = list(rand_str) + [' '] \
* (max_char - len(rand_str))
for i,t in enumerate(label):
label[i] = char_list.index(t)
label = np.uint8(label)
images.append(img)
labels.append(label)
count += 1
if count == data_count:
break
else:
continue
break
else:
continue
break
else:
continue
break
end_time = dt.datetime.Now()
print("time taken to generate data",end_time - start_time)
return images,labels
引发的错误在第for f in font:
行
我在这里做错了什么?我必须使用range()吗?
解决方法
font=cv2.FONT_HERSHEY_SIMPLEX
for f in font:
...
在CV2中,字体是一个代表字体本身的简单整数,我不是完全确定(a)为何要尝试对其进行迭代。
如果要遍历字体的大小,则必须使用(例如)fontScale
的{{1}}参数。
如果要遍历字体的集合,则必须提供该集合,例如其中之一:
putText()
如果只有 one 字体,则根本不要对其进行迭代。摆脱font = [cv2.FONT_HERSHEY_SIMPLEX] # one font as a collection
font = [cv2.FONT_HERSHEY_SIMPLEX,cv2.FONT_HERSHEY_PLAIN] # two fonts
行(取消缩进当前“里面”的内容),只要在当前使用for f in font
的任何地方使用font
。
(a) Python在尝试弄清您的意图时遇到了 like 麻烦:-)