search

BumbleBee-BBStream / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
0 stars 0 forks source link

Add support for combining multiple datasets for training #33

Closed gaoyuan closed 7 months ago

gaoyuan commented 7 months ago

Now we can pass in multiple dataset ids as the following:

docker exec -it yolov7_pytorch_devel python3 auto_train.py --dataset-ids 667430116996943872 667397486096158720 --project-path /home/workspace/BatteryDataSet --project-version 14_defect_5_JinChuan_V7

It's backward compatible so passing only 1 dataset id is also fine.

The datasets will be downloaded in parallel.