专栏算法工具链在OE包中训练focs模型

在OE包中训练focs模型

已解决
ZhongJJ2022-12-06
54
2

请问如何在OE包中训练自己的fcos模型?个人的数据集的排布是遵从什么样的排布呢?如何将自己的数据集转换成lmdb格式?谢谢

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2

    在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中配置

    2022-12-07
    0
    1
    • ZhongJJ回复颜值即正义:

      了解,感谢您的回答。

      2022-12-07
      0