Skip to content

训练方法咨询 #112

@zhjygit

Description

@zhjygit

师傅你好,按照项目里面的训练方法,我下载了预训练权重,能识别的类型是:aeroplane bicycle bird boat bottle bus car cat
chair cow diningtable dog horse motorbike person pottedplant sheep sofa train tvmonitor 。
我像识别的目标是工业安全帽helmet,将其添加到最后一个。然后准备了1500张左右的安全帽图片,并对xml文件、jpg等进行了手动整理,voc_annotation.py运行正常。
接着运行train.py:
image
简单做了训练测试,gpu是16G显存,占用率30%左右,但迭代速率很低只有1.5左右。训练20分钟左右后,将新的权重加载并预测。
预测结果非常郁闷的是,安全帽识别正常,但半身人像也被识别为了安全帽:
image
我的数据集也手动清理过无关的照片,整体没有单独的半身像:
image
非常郁闷,到底是哪里出问题了呢

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions