alexgkendall / SegNet-Tutorial

Files for a tutorial to train SegNet for road scenes using the CamVid dataset
http://mi.eng.cam.ac.uk/projects/segnet/tutorial.html
847 stars 517 forks source link

Training output perform much worse than pre-train segnet_basic_camvid model #83

Open miketry0313 opened 7 years ago

miketry0313 commented 7 years ago

Hi All,

I am fine tuning segnet_basic_camvid.caffemodel on Cityscape data by using segnet_basic_train.prototxt and segnet_basic_solver.prototxt. The output of loss and avg accuracy looks good (loss: 0.76, avg accuracy: 0.6). But when I using test_segmentation_camvid.py to display inference model result, it looks much worse than pre-train model. Can anyone give me some help? Thanks in advanced.

ground_truth (Ground Truth)

infer_result_pretrain_model ( Pretrain Model Inference Result)

Below are the result of training output model before calculating bn statistics and after bn statistics. infer_result_before_compute_stat

infer_result_after_compute_stat