AttributeError: 'list' object has no attribute 'view' about yolov9 HOT 8 CLOSED

YOLOv9 models should be trained with train_dual.py

from yolov9.

The details of GELAN and YOLOv9 are shown in the paper. Simply to say, YOLOv9 models contain additional aux branch for training. Our experiments are conduct with train-from-scratch strategy, so there are no pre-trained weights. If you want to fine-tune yolov9 on your dataset with train_dual.py, please use yolov9-c.pt.

from yolov9.

In the train_dual.py, if '--weights' = 'yolov9-c.pt' , is '--cfg' = 'yolov9-c.yaml' ?

from yolov9.

What is the difference between train.py and train_dual.py In terms of using training methods ?

from yolov9.

#1 (comment)

from yolov9.

In general.py: device = prediction.device
The error is as follows：
AttributeError: 'list' object has no attribute 'device'

from yolov9.

For inference, please check #11 (comment).

from yolov9.

What is the difference between yolov9-c.pt and gelan-c.pt?
In the train_dual.py, which weight file is commonly used to load pre training weights?

from yolov9.