nickliqian / darknet_captcha

基于darknet实现目标检测,提供识别点选验证码的实例和训练自己数据的API
Apache License 2.0
398 stars 128 forks source link

建议作者修改一下readme文件 #21

Open 462421714 opened 2 years ago

462421714 commented 2 years ago

主要有两处错误:

  1. 在第5步 生成样本 之后需要修改my_captcha.name中的标签值,单类型目标检测需改为word。 多类型目标检测,需改为word dummy。 否则,在output_label.py中第33行:

    if cls not in classes or int(difficult) == 1:
        continue

    会判断标签值不在classes中,从而labels_data中的txt文件为空。 在后续训练不会报错,但训练结果一直为: Avg IOU: -nan, Class: -nan, Obj: -nan, No Obj: 0.001772, .5R: -nan, .75R: -nan, count: 0

  2. 第6步 划分训练集和验证集 代码应为: python3 extend/output_label.py my_captcha 无需参数1