You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
More specific, I trained the model with my custom dataset and saved the weight. But when it's load the weight and predict on the same image, the difference result after each time I run. I think it should have some random factor in here but I can't find any? So can you explain that?
You have to change this line of code in model.py
line 133: cnn_model = CNN(self.img_data, True) to cnn_model = CNN(self.img_data, not self.forward only)
Batch norm is the cause of this problem.
More specific, I trained the model with my custom dataset and saved the weight. But when it's load the weight and predict on the same image, the difference result after each time I run. I think it should have some random factor in here but I can't find any? So can you explain that?
command: python3 src/launcher.py --phase=test --visualize --data-path=data/test_1/test.txt --data-base-dir=./ --log-path=log.txt --load-model --model-dir=train/ --output-dir=results
The text was updated successfully, but these errors were encountered: