在OE包中训练focs模型

请问如何在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中配置

了解,感谢您的回答。