lufficc/ssd训练自己数据集
时间: 2023-10-05 20:09:47 浏览: 125
lufficc/ssd是一个用于目标检测的开源项目,你可以使用它来训练自己的数据集。你可以按照以下步骤进行操作:
1. 下载lufficc/ssd代码,你可以在某乎上的一个博主提供的链接中找到相应的代码。
2. 确保你已经准备好了自己的数据集,并将其转换为SSD网络所需的格式。你可以参考参考2中的方法来实现这一步骤。
3. 需要注意的是,在训练之前,你需要为每个图像生成对应的ground truth(GT)。你可以参考参考3中的方法来实现这一步骤。
4. 在训练之前,你可以根据需要调整defaults.py中的_C.MODEL.BACKBONE.PRETRAINED参数,如果你有自己的预训练权重文件,你可以将其放在对应的文件夹下,并将其命名为--mobilenet_v2-b0353104.pth,这样你就可以直接使用自己训练好的权重进行训练了。
5. 运行train.py文件进行训练。
阅读全文