search

IanTaehoonYoo / semantic-segmentation-pytorch

Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.
MIT License
86 stars 19 forks source link

FCN8 object has no attribute 'img_width' #16

Closed rkuo2000 closed 3 years ago

rkuo2000 commented 3 years ago

predict.py ---> 81 model_width = model.img_width 82 model_height = model.img_height

ModuleAttributeError: 'FCN8' object has no attribute 'img_width'

rkuo2000 commented 3 years ago

The cause is I directly use pretrain model to predict without training the pretrain model. model_name = "pspnet_mobilenet_v2" has much better result than "fcn8_vgg16" rescheduler did not help trainer (it was commented anyway)

https://kaggle.com/rkuo2000/semantic-segmentation-pytorch for your reference !