We use the python=3.8.17,pytorch =2.0.1 and cuda=11.8.
If you want to use OD-YOLO, please pip ultralytics.
After pip is finished, please replace the ultralytics you just installed with this one.
The OD-YOLO network is shown as yolov8n_OD-YOLO.yaml.
If you want to see the code for OD-YOLO's components, you can open the python file under ultralytics/nn/modules.
For ultralytics dataset training format you can refer to this URL https://docs.ultralytics.com/.
You can use command in the terminal:
yolo detect train data=yourdataset model=yolov8n_OD-YOLO.yaml epochs=XXXX batch=XXXXX