search

haooyuee / YOLOv5-AUC-ROC-MedDetect

Implement auroc metric on yolov5
GNU Affero General Public License v3.0
4 stars 1 forks source link

AUR curve does not exist #2

Open EslamHaourn opened 2 months ago

EslamHaourn commented 2 months ago

Hi, Thanks for your great repo. After I finished my training, I didn't find the roc curve. what should I do? Thanks in advance.

haooyuee commented 2 months ago

Hi, I've set the save path of the auroc curve to be the same as the other metrics in the original version. You can try to find it based on this issue case. Let me know your progress. https://github.com/ultralytics/yolov5/issues/10365

EslamHaourn commented 2 months ago

Hello,
Thanks for your response. But I still can not find the roc curve. Could you write the code edit please? Does this repo compute the roc curve? https://github.com/haooyuee/yolov5-auc thank you

haooyuee commented 2 months ago

aucroc.plot_polar_chart(auc_scores, save_dir=save_dir, names=list(names.values()))

Please comment out this line in the val.py file. There is a bug in the code for saving plot_polar_chart, causing the process get stuck. I will fix this bug asap

EslamHaourn commented 2 months ago

Hi, I will try this. Concerning the anchor boxes, when I change it in the .yaml file, the training process read the old values. Thanks in advance.

haooyuee commented 2 months ago
  1. I have fixed the bug in function plot_polar_chart.
  2. You may need set correct configurations : --cfg Path_to_your_new_yaml_file to let train.py find it.
  3. Changing the anchors box may greatly affect the performance of YOLO. The current anchors settings already cover three types of bbox: small, medium and large. I suggest you go to the issue of official yolov5 repo to ask this question.
EslamHaourn commented 2 months ago

Hello, I commented out the line you mentioned and I have found the roc curve. I already have set the model.yaml but the training process still read a different anchors. Thanks.

qjl1244281167 commented 3 weeks ago

No pred db for cls 2, Set the auc value to 0 ...