Пример #1
0
 def setUp(self):
     torch.manual_seed(43)
     self.model = get_model_no_weights(self.CONFIG_PATH)
Пример #2
0
 def setUp(self):
     self.model = get_model_no_weights(
         "COCO-Detection/retinanet_R_50_FPN_1x.yaml")
Пример #3
0
 def setUp(self):
     self.model = get_model_no_weights(
         "COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml")
Пример #4
0
 def setUp(self):
     self.model = get_model_no_weights(
         "COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml")