Spaces:

AndreasLH
/

Weak-Cube-RCNN

Running

AndreasLH commited on May 19

Commit

9eee6dd

1 Parent(s): d7ee9f5

no pretrain to avoid download of model

Files changed (1) hide show

cubercnn/modeling/backbone/dla.py CHANGED Viewed

@@ -493,7 +493,7 @@ def build_dla_from_vision_fpn_backbone(cfg, input_shape: ShapeSpec, priors=None)
     imagenet_pretrain = cfg.MODEL.WEIGHTS_PRETRAIN + cfg.MODEL.WEIGHTS == ''
-    bottom_up = DLABackbone(cfg, input_shape, pretrained=imagenet_pretrain)
     in_features = cfg.MODEL.FPN.IN_FEATURES
     out_channels = cfg.MODEL.FPN.OUT_CHANNELS

     imagenet_pretrain = cfg.MODEL.WEIGHTS_PRETRAIN + cfg.MODEL.WEIGHTS == ''
+    bottom_up = DLABackbone(cfg, input_shape, pretrained=False)
     in_features = cfg.MODEL.FPN.IN_FEATURES
     out_channels = cfg.MODEL.FPN.OUT_CHANNELS