I have trained the model with resnet18.
I have given 2 classes.
when i run test.py. After 10 epochs
I am getting something like this…
INFO:root:Writing pedestrian VOC results file
INFO:root:Writing bicycle VOC results file
INFO:root:AP for background = 0.5177
INFO:root:AP for pedestrian = 0.2642
INFO:root:Mean AP = 0.3910
Unable to detect 2nd class.
What is the problem??
There is something wrong in the train.py?
Please clarify this…
Thanks in advance.