@@ -19,7 +19,7 @@ def test_load_model(self):
19
19
model_path = MmdetTestConstants .MMDET_CASCADEMASKRCNN_MODEL_PATH ,
20
20
config_path = MmdetTestConstants .MMDET_CASCADEMASKRCNN_CONFIG_PATH ,
21
21
confidence_threshold = 0.3 ,
22
- device = None ,
22
+ device = "cpu" ,
23
23
category_remapping = None ,
24
24
load_at_init = True ,
25
25
)
@@ -36,7 +36,7 @@ def test_perform_inference_with_mask_output(self):
36
36
model_path = MmdetTestConstants .MMDET_CASCADEMASKRCNN_MODEL_PATH ,
37
37
config_path = MmdetTestConstants .MMDET_CASCADEMASKRCNN_CONFIG_PATH ,
38
38
confidence_threshold = 0.5 ,
39
- device = None ,
39
+ device = "cpu" ,
40
40
category_remapping = None ,
41
41
load_at_init = True ,
42
42
)
@@ -74,7 +74,7 @@ def test_perform_inference_without_mask_output(self):
74
74
model_path = MmdetTestConstants .MMDET_RETINANET_MODEL_PATH ,
75
75
config_path = MmdetTestConstants .MMDET_RETINANET_CONFIG_PATH ,
76
76
confidence_threshold = 0.5 ,
77
- device = None ,
77
+ device = "cpu" ,
78
78
category_remapping = None ,
79
79
load_at_init = True ,
80
80
)
@@ -110,7 +110,7 @@ def test_convert_original_predictions_with_mask_output(self):
110
110
model_path = MmdetTestConstants .MMDET_CASCADEMASKRCNN_MODEL_PATH ,
111
111
config_path = MmdetTestConstants .MMDET_CASCADEMASKRCNN_CONFIG_PATH ,
112
112
confidence_threshold = 0.5 ,
113
- device = None ,
113
+ device = "cpu" ,
114
114
category_remapping = None ,
115
115
load_at_init = True ,
116
116
)
@@ -157,7 +157,7 @@ def test_convert_original_predictions_without_mask_output(self):
157
157
model_path = MmdetTestConstants .MMDET_RETINANET_MODEL_PATH ,
158
158
config_path = MmdetTestConstants .MMDET_RETINANET_CONFIG_PATH ,
159
159
confidence_threshold = 0.5 ,
160
- device = None ,
160
+ device = "cpu" ,
161
161
category_remapping = None ,
162
162
load_at_init = True ,
163
163
)
@@ -200,7 +200,7 @@ def test_create_original_predictions_from_object_prediction_list_with_mask_outpu
200
200
model_path = MmdetTestConstants .MMDET_CASCADEMASKRCNN_MODEL_PATH ,
201
201
config_path = MmdetTestConstants .MMDET_CASCADEMASKRCNN_CONFIG_PATH ,
202
202
confidence_threshold = 0.5 ,
203
- device = None ,
203
+ device = "cpu" ,
204
204
category_remapping = None ,
205
205
load_at_init = True ,
206
206
)
@@ -245,7 +245,7 @@ def test_create_original_predictions_from_object_prediction_list_without_mask_ou
245
245
model_path = MmdetTestConstants .MMDET_RETINANET_MODEL_PATH ,
246
246
config_path = MmdetTestConstants .MMDET_RETINANET_CONFIG_PATH ,
247
247
confidence_threshold = 0.5 ,
248
- device = None ,
248
+ device = "cpu" ,
249
249
category_remapping = None ,
250
250
load_at_init = True ,
251
251
)
0 commit comments