YunYang1994/tensorflow-yolov3

训练自己的数据集,并使用训练好的模型进行测试(少走弯路)

Open

#588 opened on 2020年11月30日

GitHub で見る
 (8 comments) (3 reactions) (0 assignees)Python (1,380 forks)batch import
good first issue

Repository metrics

Stars
 (3,630 stars)
PR merge metrics
 (30d に merged PR はありません)

説明

    1. 准备数据,制作训练时需要的文件(data文件夹下面的四个文件:数据类名,锚框,训练数据,测试数据)
    1. 根据步骤1修改core/config.py文件
    1. 基于yolo原始权重训练,(或者从头训练)(这步教程写的很清楚)
    1. 当loss为1左右时,模型就已经有效果了,训练过程中会保存很多模型,选择一个loss最小的,并根据所选择的模型名修改freeze_graph.py,修改后运行freeze_graph.py,会在项目根目录生成***.pb文件
    1. 修改demo文件,如修改video_demo.py,修改调用的***.pb文件路径,修改num_class为自己数据类个数

コントリビューターガイド