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:
|