训练自己的数据集

[lsl0901]

作者您好！！十分感激您的工作和努力！！
我想训练自己的数据集，我该如何安排和重构我的数据集呢，或者说我需要生成哪些文件，他们的构成又是什么样的。
再次感谢！！

[YuHengsss]

您好，关于视频数据的标注可以参考 OVIS 数据集的格式（只标注box）或者 Argoverse 的格式。我们代码里提供了将它们转化为coco标注格式以及数据加载的pipline：

YOLOV/yolox/data/datasets/vid.py

Line 204 in 8fe37e4

class Arg_VID(torchDataset):

YOLOV/yolox/data/datasets/vid.py

Line 414 in 8fe37e4

class OVIS(Arg_VID):

YOLOV/yolox/data/datasets/ovis.py

Line 238 in 8fe37e4

def convert_ovis_coco(data_dir = '/opt/dataset/OVIS/annotations_train.json',save_dir = '/opt/dataset/OVIS/ovis_train.json'):

[lsl0901]

@YuHengsss
感谢您的回复！我会尝试您的建议。