VOC Classification Code is not working for my model that has the alexnet structure about voc-classification HOT 2 OPEN

Hello,

I am facing the same problem. I am trying to finetune my model on Pascal VOC 2007 dataset and the loss is not going down beyond random. It's not clear what is going wrong.

I have tried with default hyperparameter settings as well as my own.

Please advise.

Thanks

from voc-classification.

I encountered the same problem as @Maryames . And I solved it with the help of @uahsan3 .
If anyone else encounter this problem, please look at the trainval.prototxt generate by the code (in train_cls.py)and check the data layer. The code will create a right data layer and if your alexnet.prototxt also include a data layer, then the network will have two data layer and the image data will not go through the network. Besides, I attach two screenshots of trainval.prototxt below, the first is wrong and the second is right.

from voc-classification.