Jim*_*ela -2 python python-3.7
我正在尝试使用不同的字体为 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), f, fs, \
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
Run Code Online (Sandbox Code Playgroud)
引发的错误在行: for f in font:
我在这里做错了什么?我必须使用 range() 吗?
font=cv2.FONT_HERSHEY_SIMPLEX
for f in font:
...
Run Code Online (Sandbox Code Playgroud)
在 CV2 中,字体是一个简单的整数,代表字体本身,我不完全确定(a)为什么要迭代它。
如果你想遍历大小的字体,你将不得不使用(例如)的fontScale参数putText()。
如果要迭代一组字体,则必须提供该集合,例如使用以下之一:
font = [cv2.FONT_HERSHEY_SIMPLEX] # one font as a collection
font = [cv2.FONT_HERSHEY_SIMPLEX, cv2.FONT_HERSHEY_PLAIN] # two fonts
Run Code Online (Sandbox Code Playgroud)
如果您只有一种字体,则根本不要对其进行迭代。摆脱该for f in font行(取消缩进当前“内部”的内容),只需font在您当前使用的任何地方使用f.
(a) Python 在试图弄清楚你的意图时遇到了类似的麻烦:-)
| 归档时间: |
|
| 查看次数: |
228 次 |
| 最近记录: |