search

rosinality / ocr-pytorch

Object-Contextual Representations for Semantic Segmentation in PyTorch
MIT License
63 stars 14 forks source link

backbone #6

Open Youzhihui opened 3 years ago

Youzhihui commented 3 years ago

Hello Rosinality, I can not find the "vovnet39_torchvision.pth" file, Can you share the weight file? And if I can take resnet101 as the backbone? How to do thanks

rosinality commented 3 years ago

You can use checkpoints in here: https://github.com/stigma0617/VoVNet.pytorch. And if you want to use resnet101, then you need to modify resnet code to return stride 16, 32 feature maps so it could be used in here: https://github.com/rosinality/ocr-pytorch/blob/master/model.py#L51.