IanTaehoonYoo / semantic-segmentation-pytorch

Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FCN8 object has no attribute 'img_width'

rkuo2000 opened this issue · comments

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

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

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 !