Spaces:
Runtime error
Runtime error
prathmeshrmadhu
commited on
Commit
·
2cd210b
1
Parent(s):
17b96f6
trying to send the model
Browse files- model.py +4 -3
- model_dict/detection.yaml +1 -4
- pretrained_models/fasterrcnn-odor-ep100.pth +3 -0
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 |
-
|
39 |
-
|
|
|
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
|