PlaneNet icon indicating copy to clipboard operation
PlaneNet copied to clipboard

Failed to load pretrained deeplab model for training

Open EryiXie opened this issue 5 years ago • 1 comments

Hi author,

thanks for your impressive work. Anyhow I met an issue by run train_planenet.py with a pretrained deeplab model. The model is downloaded from this page https://github.com/DrSleep/tensorflow-deeplab-resnet under the chapter "Caffe to TensorFlow conversion"

The error which I got looks like: [[Node: save_1/RestoreV2_310 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save_1/Const_0_0, save_1/RestoreV2_310/tensor_names, save_1/RestoreV2_310/shape_and_slices)]] 2019-01-21 16:31:31.072749: W tensorflow/core/framework/op_kernel.cc:1192] Not found: Tensor name "crfrnn/compatibility_matrix" not found in checkpoint files /home/xie/python_nn_framework/PlaneNet-master/pretrainedModels/deeplab_resnet.ckpt

It would be great if anyone can figure out, why error like this appears or did I make a mistake somewhere somehow? Thanks.

EryiXie avatar Jan 21 '19 15:01 EryiXie

Sorry for the late response. I just fixed the bug. Please let me know if it still doesn't work.

chenliu-wustl avatar Jan 30 '19 21:01 chenliu-wustl