search

DrSleep / tensorflow-deeplab-resnet

DeepLab-ResNet rebuilt in TensorFlow
MIT License
1.25k stars 429 forks source link

Finetune with your kitti example error? #100

Closed JackHenry1992 closed 7 years ago

JackHenry1992 commented 7 years ago

I follow your step of kitti example (https://gist.github.com/DrSleep/4bce37254c5900545e6b65f6a0858b9c#file-masks_kitti-ipynb) , but I got error when run finetune.py,

image

Seems that numclasses = 14 don't match with your pre-trained model. What should I do to resolve this problem?

Thank you very much for sharing this greate job!

JackHenry1992 commented 7 years ago

Solve this problem by setting --not-restore-last (python finetune.py --not-restore-last)