GJ_git
GJ_git
我自己写了个单独测试的代码 import models if __name__ == '__main__': resume = 'RCFcheckpoint_epoch12.pth' model = models.RCF() model.eval() checkpoint = torch.load(resume,map_location='cpu') try: model.load_state_dict(checkpoint) except Exception: model.load_state_dict(checkpoint['state_dict'],False) img_path = r'D:\Work\line_extraction\RCF-pytorch-master\BSR\BSDS500\data\images\train\12003.jpg' result_path = './examples/demo.png' original_img =...
I can download resnext-101 model from http://data.mxnet.io/models/imagenet/resnext/101-layers/resnext-101-0000.params