class imageEncoder(nn.Module):
def __init__(self, args):
super().__init__()
self.args = args
self.resnet50 = cv_models.resnet50()
#print(self.resnet50)
load_resnet(args.cv_model_path, self.resnet50)
# args.remove_relu = True
python-简单的resnet特征抽取器
于 2021-08-21 14:15:20 首次发布