shvardhan commited on
Commit
e02bc83
·
1 Parent(s): 58d1469

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 = 'https://github.com/open-mmlab/mmdetection/tree/master/configs/faster_rcnn/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:
 
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: