search

warmspringwinds / pytorch-segmentation-detection

Image Segmentation and Object Detection in Pytorch
MIT License
746 stars 170 forks source link

fully_conv in vgg16 #4

Closed IssamLaradji closed 6 years ago

IssamLaradji commented 6 years ago

Very good repository.

How did you make this work ? it seems that vgg16 does not have the fully_conv keyword in torchvision 

vgg16 = models.vgg16(pretrained=True,
                             fully_conv=True)
IssamLaradji commented 6 years ago

oh nvm, you have a forked version of torchvision issue solved!