prathmeshrmadhu commited on
Commit
2cd210b
1 Parent(s): 17b96f6

trying to send the model

Browse files
model.py CHANGED
@@ -34,9 +34,10 @@ def _update_model_dict_if_hf_token_is_given(
34
  for dic in model_dict.values():
35
  ckpt_path = dic['model']
36
  name = ckpt_path.split('/')[-1]
37
- ckpt_path = huggingface_hub.hf_hub_download('hysts/mmdetection',
38
- f'models/{name}',
39
- use_auth_token=token)
 
40
  dic['model'] = ckpt_path
41
 
42
 
 
34
  for dic in model_dict.values():
35
  ckpt_path = dic['model']
36
  name = ckpt_path.split('/')[-1]
37
+ # ckpt_path = huggingface_hub.hf_hub_download('hysts/mmdetection',
38
+ # f'models/{name}',
39
+ # # use_auth_token=token)
40
+ ckpt_path = "pretrained_models/fasterrcnn-odor-ep100.pth"
41
  dic['model'] = ckpt_path
42
 
43
 
model_dict/detection.yaml CHANGED
@@ -1,6 +1,3 @@
1
  Faster R-CNN (R-50-FPN):
2
  config: https://github.com/open-mmlab/mmdetection/tree/master/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.py
3
- model: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_2x_coco/faster_rcnn_r50_fpn_2x_coco_bbox_mAP-0.384_20200504_210434-a5d8aa15.pth
4
- Faster R-CNN (X-101-64x4d-FPN):
5
- config: https://github.com/open-mmlab/mmdetection/tree/master/configs/faster_rcnn/faster_rcnn_x101_64x4d_fpn_1x_coco.py
6
- model: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_x101_64x4d_fpn_1x_coco/faster_rcnn_x101_64x4d_fpn_1x_coco_20200204-833ee192.pth
 
1
  Faster R-CNN (R-50-FPN):
2
  config: https://github.com/open-mmlab/mmdetection/tree/master/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.py
3
+ model: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_2x_coco/faster_rcnn_r50_fpn_2x_coco_bbox_mAP-0.384_20200504_210434-a5d8aa15.pth
 
 
 
pretrained_models/fasterrcnn-odor-ep100.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e40b49ad1afe524e759a684616aeccd8ab592d980ff5c5045035ecbc6a76542
3
+ size 333521191