Faster RCNN on tx2

I used this web( to train faster rcnn(I input command line" python ./tools/ --gpu 0 " on terminal). It has a problem below the picture. How to solve it?


It seems the dimension doesn’t align in the ROI layer.

A common issue is that you have a different class number compared to the pre-train model.
You can remove the final fully connected layer (fc8?) and retrain it from a random initial directly.