edgeyolo 如果我有数据集，可以用您的模型训练一下吗？

如果我有数据集，可以用您的模型训练一下吗？如何操作 | edgeyolo.train.launch_train:load_train_settings:31 - param 'model_cfg' not list in settings file, use default value: model_cfg=params/model/edgeyolo.yaml | edgeyolo.train.launch_train:load_train_settings:31 - param 'weights' not list in settings file, use default value: weights=output/train/edgeyolo_coco/last.pth | edgeyolo.train.launch_train:load_train_settings:31 - param 'use_cfg' not list in settings file, use default value: use_cfg=False | edgeyolo.train.launch_train:load_train_settings:31 - param 'output_dir' not list in settings file, use default value: output_dir=output/train/edgeyolo_coco | edgeyolo.train.launch_train:load_train_settings:31 - param 'save_checkpoint_for_each_epoch' not list in settings file, use default value: save_checkpoint_for_each_epoch=True | edgeyolo.train.launch_train:load_train_settings:31 - param 'log_file' not list in settings file, use default value: log_file=log.txt | edgeyolo.train.launch_train:load_train_settings:31 - param 'dataset_cfg' not list in settings file, use default value: dataset_cfg=params/dataset/coco.yaml | edgeyolo.train.launch_train:load_train_settings:31 - param 'batch_size_per_gpu' not list in settings file, use default value: batch_size_per_gpu=8 | edgeyolo.train.launch_train:load_train_settings:31 - param 'loader_num_workers' not list in settings file, use default value: loader_num_workers=4 | edgeyolo.train.launch_train:load_train_settings:31 - param 'num_threads' not list in settings file, use default value: num_threads=1 | edgeyolo.train.launch_train:load_train_settings:31 - param 'device' not list in settings file, use default value: device=[0, 1, 2, 3] | edgeyolo.train.launch_train:load_train_settings:31 - param 'fp16' not list in settings file, use default value: fp16=False | edgeyolo.train.launch_train:load_train_settings:31 - param 'cudnn_benchmark' not list in settings file, use default value: cudnn_benchmark=False | edgeyolo.train.launch_train:load_train_settings:31 - param 'optimizer' not list in settings file, use default value: optimizer=SGD | edgeyolo.train.launch_train:load_train_settings:31 - param 'max_epoch' not list in settings file, use default value: max_epoch=300 | edgeyolo.train.launch_train:load_train_settings:31 - param 'close_mosaic_epochs' not list in settings file, use default value: close_mosaic_epochs=15 | edgeyolo.train.launch_train:load_train_settings:31 - param 'lr_per_img' not list in settings file, use default value: lr_per_img=0.00015625 | edgeyolo.train.launch_train:load_train_settings:31 - param 'warmup_epochs' not list in settings file, use default value: warmup_epochs=5 | edgeyolo.train.launch_train:load_train_settings:31 - param 'warmup_lr_ratio' not list in settings file, use default value: warmup_lr_ratio=0.0 | edgeyolo.train.launch_train:load_train_settings:31 - param 'final_lr_ratio' not list in settings file, use default value: final_lr_ratio=0.05 | edgeyolo.train.launch_train:load_train_settings:31 - param 'loss_use' not list in settings file, use default value: loss_use=['bce', 'bce', 'giou'] | edgeyolo.train.launch_train:load_train_settings:31 - param 'input_size' not list in settings file, use default value: input_size=[640, 640] | edgeyolo.train.launch_train:load_train_settings:31 - param 'multiscale_range' not list in settings file, use default value: multiscale_range=5 | edgeyolo.train.launch_train:load_train_settings:31 - param 'weight_decay' not list in settings file, use default value: weight_decay=0.0005 | edgeyolo.train.launch_train:load_train_settings:31 - param 'momentum' not list in settings file, use default value: momentum=0.9 | edgeyolo.train.launch_train:load_train_settings:31 - param 'enhance_mosaic' not list in settings file, use default value: enhance_mosaic=True | edgeyolo.train.launch_train:load_train_settings:31 - param 'use_ema' not list in settings file, use default value: use_ema=True | edgeyolo.train.launch_train:load_train_settings:31 - param 'enable_mixup' not list in settings file, use default value: enable_mixup=True | edgeyolo.train.launch_train:load_train_settings:31 - param 'mixup_scale' not list in settings file, use default value: mixup_scale=[0.5, 1.5] | edgeyolo.train.launch_train:load_train_settings:31 - param 'mosaic_scale' not list in settings file, use default value: mosaic_scale=[0.1, 2.0] | edgeyolo.train.launch_train:load_train_settings:31 - param 'flip_prob' not list in settings file, use default value: flip_prob=0.5 | edgeyolo.train.launch_train:load_train_settings:31 - param 'mosaic_prob' not list in settings file, use default value: mosaic_prob=1 | edgeyolo.train.launch_train:load_train_settings:31 - param 'mixup_prob' not list in settings file, use default value: mixup_prob=1 | edgeyolo.train.launch_train:load_train_settings:31 - param 'degrees' not list in settings file, use default value: degrees=10 | edgeyolo.train.launch_train:load_train_settings:31 - param 'hsv_gain' not list in settings file, use default value: hsv_gain=[0.0138, 0.664, 0.464] | edgeyolo.train.launch_train:load_train_settings:31 - param 'eval_at_start' not list in settings file, use default value: eval_at_start=False | edgeyolo.train.launch_train:load_train_settings:31 - param 'val_conf_thres' not list in settings file, use default value: val_conf_thres=0.001 | edgeyolo.train.launch_train:load_train_settings:31 - param 'val_nms_thres' not list in settings file, use default value: val_nms_thres=0.65 | edgeyolo.train.launch_train:load_train_settings:31 - param 'eval_only' not list in settings file, use default value: eval_only=False | edgeyolo.train.launch_train:load_train_settings:31 - param 'obj_conf_enabled' not list in settings file, use default value: obj_conf_enabled=True | edgeyolo.train.launch_train:load_train_settings:31 - param 'eval_interval' not list in settings file, use default value: eval_interval=1 | edgeyolo.train.launch_train:load_train_settings:31 - param 'print_interval' not list in settings file, use default value: print_interval=100 | edgeyolo.train.launch_train:load_train_settings:31 - param 'load_optimizer_params' not list in settings file, use default value: load_optimizer_params=True | edgeyolo.train.launch_train:load_train_settings:31 - param 'train_backbone' not list in settings file, use default value: train_backbone=True | edgeyolo.train.launch_train:load_train_settings:31 - param 'train_start_layers' not list in settings file, use default value: train_start_layers=51 | edgeyolo.train.launch_train:load_train_settings:31 - param 'force_start_epoch' not list in settings file, use default value: force_start_epoch=-1 geyolo-main\train.py", line 16, in else args.cfg) geyolo-main\edgeyolo\train\launch_train.py", line 101, in launch \lib\site-packages\torch\multiprocessing\spawn.py", line 188, in start_processes \lib\site-packages\torch\multiprocessing\spawn.py", line 150, in join error_index, failed_process.pid) ocessRaisedException:

-- Process 3 terminated with the following error: Traceback (most recent call last): File "D:\ProgramData\Anaconda3\lib\site-packages\torch\multiprocessing\spawn.py", line 59, in wrap fn(i, *args) File "D:\myworkspace\yolov5\edgeyolo-main\edgeyolo\train\launch_train.py", line 50, in train_single torch.cuda.set_device(device) File "D:\ProgramData\Anaconda3\lib\site-packages\torch\cuda_init.py", line 261, in set_device torch._C._cuda_setDevice(device) RuntimeError: CUDA error: invalid device ordinal

Feb 22 '23 13:02 g101400

这个错误报的是没有读取到cfg文件的参数，请检查训练的cfg文件（示例在文件夹params/train下）位置是否填写正确

Feb 22 '23 13:02 LSH9832

具体步骤请按照README中的说明进行，目前支持COCO/VOC/DOTA/VisDrone四种数据集格式（示例在params/dataset下），如为其他格式数据集需要进行格式转换

Feb 22 '23 13:02 LSH9832