search

bowenc0221 / panoptic-deeplab

This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)
Apache License 2.0
590 stars 117 forks source link

No module named ‘torchvision.models.utils‘ errors #126

Open upright2003 opened 1 year ago

upright2003 commented 1 year ago

Hello! I found panoptic-deeplab/segmentation/model/backbone/ under the directory

hrnet.py mnasnet.py mobilenet.py resnet.py xception.py

python file

from torchvision.models.utils import load_state_dict_from_url

need to be changed to

from torch.hub import load_state_dict_from_url

This panoptic-deeplab can only be executed, can you update the code? Thank you!