YuHengsss / YOLOV

This repo is an implementation of PyTorch version YOLOV Series
Apache License 2.0
278 stars 39 forks source link

How to customize data sets for training #74

Closed returnlsz closed 4 months ago

returnlsz commented 4 months ago

Hello, author! Your work is very good! But I've read your documentation and it's still not clear how I need to train a custom dataset. As far as I know, I need to change my custom dataset folder format to the same as ILSVRC2015_VID dataset, then prepare the train.npy and val.npy sequence files, Is there anything else I need to do?

Linengyao commented 3 weeks ago

你好,作者!你的工作非常好!但是我已经阅读了您的文档,但仍然不清楚我需要如何训练自定义数据集。据我所知,我需要将自定义数据集文件夹格式更改为与数据集相同的格式ILSVRC2015_VID然后准备train.npy和val.npy序列文件,我还需要做什么吗?

请问您解决了吗,如何训练自定义数据集

returnlsz commented 3 weeks ago

把数据集按照ILSVRC2015_VID数据集的格式搭建好,然后生成对应的train_npy和val_npy即可