Image Segmentation
Transformers
PyTorch
upernet
Inference Endpoints
File size: 142 Bytes
b13b124
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
import torch

pthfile = r'/SEG/mmsegmentation/pretrained_model/recipe1MP_R50.pth'


net=torch.load(pthfile)

for k in net.keys():
    print(k)