Spaces:
Runtime error
Runtime error
update config file
Browse files
configs/_base_/faster-rcnn_r50_fpn_1x_coco.py
CHANGED
@@ -1,12 +1,6 @@
|
|
1 |
# model settings
|
2 |
model = dict(
|
3 |
type='FasterRCNN',
|
4 |
-
data_preprocessor=dict(
|
5 |
-
type='DetDataPreprocessor',
|
6 |
-
mean=[123.675, 116.28, 103.53],
|
7 |
-
std=[58.395, 57.12, 57.375],
|
8 |
-
bgr_to_rgb=True,
|
9 |
-
pad_size_divisor=32),
|
10 |
backbone=dict(
|
11 |
type='ResNet',
|
12 |
depth=50,
|
|
|
1 |
# model settings
|
2 |
model = dict(
|
3 |
type='FasterRCNN',
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
backbone=dict(
|
5 |
type='ResNet',
|
6 |
depth=50,
|
model.py
CHANGED
@@ -23,7 +23,7 @@ class Model:
|
|
23 |
|
24 |
|
25 |
def _load_model(self, name: str) -> nn.Module:
|
26 |
-
return init_detector(config = '
|
27 |
|
28 |
def set_model(self, name: str) -> None:
|
29 |
if name == self.model_name:
|
|
|
23 |
|
24 |
|
25 |
def _load_model(self, name: str) -> nn.Module:
|
26 |
+
return init_detector(config = 'configs/_base_/faster_rcnn_r50_fpn_1x_coco.py',checkpoint = 'models/orgaquant_pretrained.pth' , device=self.device)
|
27 |
|
28 |
def set_model(self, name: str) -> None:
|
29 |
if name == self.model_name:
|