CRAFT-pytorch
CRAFT-pytorch copied to clipboard
FIXED ImportError: cannot import name 'model_urls' from 'torchvision.models.vgg'
FIXED ImportError: cannot import name 'model_urls' from 'torchvision.models.vgg'
reproduce error as: use any imagefolder
python3 test.py --trained_model=craft_mlt_25k.pth --test_folder=./img
Before fix
After fix
Tested on
Python version 3.10.12 torch version 2.2.0 torchvision version 0.17.0
Addressed issue
#191
should merge this in and bump the requirements.txt