[CRAFT-pytorch] ImportError: cannot import name 'model_urls' from 'torchvision.models.vgg'

OCR 모델을 위한 환경 셋팅 중 직면한 문제.

원인은 torchvision 버전이 업그레이드가 되면서 'model_urls'가 사라졌다.

 

따라서 해결 방법은 간단하다. 'model_urls' 부분을 삭제해주는 것!

 

1. vgg16_bn.py에 들어간다.
2. 'model_urls' 부분을 삭제한다.

from torchvision import models
from torchvision.models.vgg import model_urls <<< 삭제

...

class vgg16_bn(torch.nn.Module):
    def __init__(self, pretrained=True, freeze=True):
        super(vgg16_bn, self).__init__()
        model_urls['vgg16_bn'] = model_urls['vgg16_bn'].replace('https://', 'http://') <<< 삭제

 

참고

https://github.com/clovaai/CRAFT-pytorch/issues/191

ImportError: cannot import name 'model_urls' from 'torchvision.models.vgg' · Issue #191 · clovaai/CRAFT-pytorch