search

Britefury / cutmix-semisup-seg

Semi-supervised semantic segmentation needs strong, varied perturbations
MIT License
163 stars 23 forks source link

Add hubconf.py #11

Open Ivan1248 opened 2 years ago

Ivan1248 commented 2 years ago

This adds hubconf.py so that, for example, the following can work when "Ivan1248" is replaced with "Britefury":

model = torch.hub.load('Ivan1248/cutmix-semisup-seg', 'resnet101_deeplab_imagenet', num_classes=19)

If you would like, you can accept this or make changes.