inging550 / YOLOV1-pytorch

主干网络替换为了改进的ResNet50
19 stars 4 forks source link

Have you test your code in PASCAL VOC and calculate mAP? #3

Closed xiazhi1 closed 7 months ago

xiazhi1 commented 7 months ago

Now, I wonder your code's performance in PASCAL VOC . I have tested in PASCAL VOC 2007 , it shows that your code's mAP is just around 52%, If I want to increse the performance , should I change the network structure?

inging550 commented 7 months ago

I haven't tested mAP, I'm just writing code for learning purposes First of all, YOLOV1 is a network many years ago, if you want to improve the performance you can try, the latest networks such as YOLOV8, RTDETR etc. If you really want to improve YOLOV1, There are many ways to potentially improve the mAP shuch as changing the network structure, data augmentation, improving training hyperparameters,replace the loss function and so on.

xiazhi1 commented 7 months ago

ok thanks a lot, I will check it