search

lxj-drifter / UIOU_files

11 stars 1 forks source link

loss.py , how to use in other models? #1

Open cheny00 opened 2 months ago

cheny00 commented 2 months ago

exp, the model is yolo. This good loss can replace the original IOU-loss? how do it? Look forward to your reply!

------If your detection model is yolo, this will be very easy, just find the destination file and replace the corresponding function. The experiments in the paper are based on yolo.

cheny00 commented 1 month ago

Thank you for your help. YOLOv5 can be directly replaced and used. However, I plan to modify IOUloss on the YOLOX model, which does not contain the contents of the two files you provided. Is there any way to port and modify it?

qianye-qiu commented 13 hours ago

您好，您知道如何使用uiou来替换掉ostrack的损失函数吗？