定义Backbone

 

5c127e78c8db4943a5a2aa61bf1c3712.png

 

 定义Neck

aacd89fc4d6d4b5fbd68bd58a5a57af6.png

 锚框的设定

主要修改num_classes为自己数据集类别。

1b4d471a37974948ba9793977d34d09b.png

 损失函数的组成

a037528e540948bfb0382896f344a002.png

 目标检测标注信息

d1bfe2af63ca499281130d508d4cee33.png

 ed61a16ff3fd44cd84f2dc71509ea1fa.png

706ccd7625a447a482a6e7169620e4a4.png 

190e02d2fa214641b1a3c3ec804a7940.png

f587a759cb914c988666e1d99d467ed0.png

 

 训练输入数据定义

sample_per_gpu是输入每个GPU的batch size。

workers_per_gpu是每个GPU分配的CPU线程数。

89839f086c8f49298e24a8356c644734.png 自定义输入数据

 9153ff10199b43bcbb0cd87da3aacc66.png

 数据增强

9802771655ca40fa8524195401b71077.png

 

9ffc5968f7b94cfd973a5ab221ea143e.png

 训练策略28187014661f4cf7999b675259eee824.png

 训练自己数据集总结fb14bc0d1d4a44468e97e6a58761f953.png

 

 

# 深度学习 # 人工智能 # 计算机视觉 # 目标检测 # 神经网络
Logo
技术共进，成长同行——讯飞AI开发者社区

技术共进，成长同行——讯飞AI开发者社区

更多推荐

cover

大模型入门书单：五本书就够了，带你从0到1，深入理解AI大模型！

avatar 讯飞AI开发者社区
cover

一 . 机器学习

avatar 讯飞AI开发者社区
cover

一场人工智能革命:Graph XAI图增强智能

avatar 讯飞AI开发者社区