CRAFT-pytorch icon indicating copy to clipboard operation
CRAFT-pytorch copied to clipboard

FIXED ImportError: cannot import name 'model_urls' from 'torchvision.models.vgg'

Open prathameshza opened this issue 1 year ago • 1 comments

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

craft_error

After fix

craft_fix

Tested on

Python version 3.10.12 torch version 2.2.0 torchvision version 0.17.0

Addressed issue

#191

prathameshza avatar Feb 12 '24 09:02 prathameshza

should merge this in and bump the requirements.txt

mattkanwisher avatar May 14 '24 11:05 mattkanwisher