fasterrcnn-pytorch-training-pipeline icon indicating copy to clipboard operation
fasterrcnn-pytorch-training-pipeline copied to clipboard

ResNet-101 network

Open samahwaleed opened this issue 2 years ago • 9 comments

  1. I want to use resnet-101 model but I can't find it in the models folder. Could you please help me!?

  2. How I can find the precision and recall for each object?

Thank you

samahwaleed avatar Jan 07 '23 02:01 samahwaleed

@samahwaleed ResNet101 model is not yet added. Will add that for sure if that is a requirement. Regarding precision and recall. You can check precision and recall, you can visualize class-wise precision and recall using the eval.py script.

sovit-123 avatar Jan 07 '23 05:01 sovit-123

I would be grateful if you could add ResNet101 as soon as possible as I need it for research purpose.

Regarding precision and recall. You can check precision and recall, you can visualize class-wise precision and recall using the eval.py script. Could you please explain more how to do it as I need these values for each class!?

Thank you for your time and consideration.

samahwaleed avatar Jan 07 '23 09:01 samahwaleed

Sure, I would add the model by tomorrow.

After training the model, use the eval.py script to check the metrics. Example command: python eval.py --weights outputs/training/custom_training/best_model.pth --config data_configs/custom_data.yaml --model fasterrcnn_resnet50_fpn_v2 --verbose

sovit-123 avatar Jan 07 '23 14:01 sovit-123

Thank you so much Sir,

python eval.py --weights outputs/training/custom_training/best_model.pth --config data_configs/custom_data.yaml --model fasterrcnn_resnet50_fpn_v2 --verbose

I have tried this and I only got the AP. How I can get the Recall for each class and F1 score?

samahwaleed avatar Jan 07 '23 16:01 samahwaleed

Right now, the library uses Torchmetrics to show class-wise mAP. I will try to add classwise recall and precision. But it may take some time.

sovit-123 avatar Jan 08 '23 00:01 sovit-123

It shows the average precision for each class but not recall, I just need the recall and F1 score.

Thank you so much.

samahwaleed avatar Jan 08 '23 01:01 samahwaleed

I can add recall per class but not sure about the F1 score. Before that, I will add the ResNet101 model.

sovit-123 avatar Jan 08 '23 01:01 sovit-123

@samahwaleed I have added the ResNet101 model.

sovit-123 avatar Jan 08 '23 02:01 sovit-123

Thank you Sir.

I would be thankful if you add the recall at the earliest possible time.

samahwaleed avatar Jan 08 '23 14:01 samahwaleed