请问如何在OE包中训练自己的fcos模型?个人的数据集的排布是遵从什么样的排布呢?如何将自己的数据集转换成lmdb格式?谢谢
在OE包中训练自己的模型:
1. 首先修改config中的配置,将device和路径配置为自己的项
2. 修改dataset为自定义dataset。数据排布可以将数据调整为coco的格式,或使用自定义的format,以下给出数据调用流程:
流程说明:所有类和接口都在config
中调用,数据加载通过调用dataloader
实现,dataloader
加载的数据集dataset由Coco
类返回,Coco
类读取的是lmdb格式的打包数据,该数据通过调用packer
接口实现,实现流程:packer
调用CocoDetectionPacker
类将CocoDetection
返回的数据(image,label…)打包成设置的格式(lmdb,mxrecord)。-
实现一个dataset需要对以上涉及的类重写,根据数据集的format重写__getitem__()
3. {stage}_trainer中配置训练策略,若基于提供的权重做训练则在float_solver中配置
了解,感谢您的回答。