jwyang / faster-rcnn.pytorch

A faster pytorch implementation of faster r-cnn
MIT License
7.71k stars 2.33k forks source link

ap get higher when validation loss added #868

Open Divenirelrn opened 3 years ago

Divenirelrn commented 3 years ago

I modify the raw code of pytorch v1.0 version to add validation loss computing capacity, here are my modifications: first, i add ground truth labels in roibatchLoader: image then, i replace if self.training with if True in two files to compute both train and validation loss: one in faster_rcnn.py: image image another in prn.py: image

When above modifications are made, i test the model trained with vgg16 and i get map of about 70% which is higher than origin inal about 65%, so here is my question, why does this happen?