imSitu icon indicating copy to clipboard operation
imSitu copied to clipboard

set the correct cnn type according to the weights filename

Open njuaplusplus opened this issue 4 years ago • 0 comments

Solve the loading problems mentioned in this closed issue: https://github.com/my89/imSitu/issues/3#issuecomment-469077352

Because class baseline_crf has a default parameter cnn_type='resenet_101', the eval.py will always try to load the weights to the resnet101 model. That's why other models can not be loaded correctly.

njuaplusplus avatar May 18 '20 13:05 njuaplusplus