定义Backbone
定义Neck
锚框的设定
主要修改num_classes为自己数据集类别。
损失函数的组成
目标检测标注信息
训练输入数据定义
sample_per_gpu是输入每个GPU的batch size。
workers_per_gpu是每个GPU分配的CPU线程数。
自定义输入数据
数据增强
训练策略![28187014661f4cf7999b675259eee824.png](https://i-blog.csdnimg.cn/blog_migrate/3c1e5c8fb56cf09d271928dcbec23b6e.png)
训练自己数据集总结![fb14bc0d1d4a44468e97e6a58761f953.png](https://i-blog.csdnimg.cn/blog_migrate/f4d8a145196abb8b45375ecc47337abe.png)
主要修改num_classes为自己数据集类别。
sample_per_gpu是输入每个GPU的batch size。
workers_per_gpu是每个GPU分配的CPU线程数。