pspnet50 pretrained model in tensorflow

[smmdream]

@hellochick could you upload the pspnet50 model which has been trained in cityscapes or ade20k

[hellochick]

Hey @smmdream , I have uploaded on google drive

[AndrewZhengXiao]

@hellochick Hi I have no idea why every time I run inference.py returns No checkpoint file found. I give the path as follow: ./model/cityscapes_model/pspnet101/ --checkpoints ./model/cityscapes_model/pspnet101/

[smmdream]

I do it in the code This is mine. You should make sure that the four files(as the follow picture) are in the path of SNAPSHOT_DIR, that's your model and the file named checkpoint will ensure which model will be loaded. such as: and ckpt-0 will be loaded.

[robin2002]

I also got the same problem. In my case, I change a file name (checkpoint) instead of checkpoint.txt. This solved my problem.

[AndrewZhengXiao]

Thanks I solved this problem as @robin2002 said. Change checkpoint.txt to checkpoint :)

[ClaireTun]

Hi~Could you plz send the pretrained model by email? I can't download from Google.😟 My email is [email protected]

[cobbwho]

I also got the same problem. In my case, I change a file name (checkpoint) instead of checkpoint.txt. This solved my problem.

Or change code in inference.py . Modify code tf.train.get_checkpoint_state(/path/to/your/dir, 'checkpoint.txt')

[bashar-tom]

Could you please upload the pretrained model for PSP50. in the google drive folder there is only the PSP101 model. Thanks in advance