Theem
/

fasterrcnn_resnet50_fpn_grayscale

Model card Files Files and versions Community

Theem commited on Mar 3, 2023

Commit

41cb296

·

1 Parent(s): d55be37

fix weights

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ The file is given as a state_dict. Thus to initialize the model run:
 # Load pretrained weights
 state_dict = torch.load(model_path, map_location=torch.device('cpu'))['model']
 # Load torchvision model
-model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights='DEFAULT')
 # Adapt input convolution
 model.backbone.body.conv1 = torch.nn.Conv2d(1, 64,
                             kernel_size=(7, 7), stride=(2, 2),

 # Load pretrained weights
 state_dict = torch.load(model_path, map_location=torch.device('cpu'))['model']
 # Load torchvision model
+model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights=None)
 # Adapt input convolution
 model.backbone.body.conv1 = torch.nn.Conv2d(1, 64,
                             kernel_size=(7, 7), stride=(2, 2),