sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
223 stars 75 forks source link

Generate the accuracy and loss figures #59

Closed samahwaleed closed 1 year ago

samahwaleed commented 1 year ago

Dear Sir,

Could you please generate the training loss and training accuracy in one figure?

Thank you

sovit-123 commented 1 year ago

@samahwaleed Hello, combining the two graphs may not help much. Because in object detection, we mostly look at the mAP graph and not the mAP graph and training loss together. The reason is that mAP is calculated on the validation set and the loss here is saved on the training set.

samahwaleed commented 1 year ago

Thank you for your kind response. I would be very thankful if you add it temporarily as I have to show this graph in the research paper.

sovit-123 commented 1 year ago

Let's see. But that does not feel very ideal. But that means that others who will use the code base will also have access to it which may cause issues.