How to add transformer-autoaugment in model

ouyanghaodong / DEYO

Implementation of paper - DEYO: DETR with YOLO for End-to-End Object Detection

GNU Affero General Public License v3.0

68 stars 5 forks source link

Open moersxm opened 6 months ago

ouyanghaodong commented 6 months ago

You can try modifying ultralytics/data/augment.py and apply it to the YOLODataset in ultralytics/data/dataset.py

moersxm commented 6 months ago

thanks