Skip to content

Commit 8254169

Browse files
committed
Merge pull request opencv#10153 from dkurt:specify_mobilenet_version
2 parents bd6a669 + cdbbccb commit 8254169

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

samples/dnn/mobilenet_ssd_accuracy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
'COCO evaluation framework is required: http://cocodataset.org')
1919
parser.add_argument('--weights', required=True,
2020
help='Path to frozen_inference_graph.pb of MobileNet-SSD model. '
21-
'Download it at https://github.com/tensorflow/models/tree/master/research/object_detection')
21+
'Download it from http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_11_06_2017.tar.gz')
2222
parser.add_argument('--prototxt', help='Path to ssd_mobilenet_v1_coco.pbtxt from opencv_extra.', required=True)
2323
parser.add_argument('--images', help='Path to COCO validation images directory.', required=True)
2424
parser.add_argument('--annotations', help='Path to COCO annotations file.', required=True)

samples/dnn/mobilenet_ssd_python.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
#
77
# Also TensorFlow model from TensorFlow object detection model zoo may be used to
88
# detect objects from 90 classes:
9-
# https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md
9+
# http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_11_06_2017.tar.gz
1010
# Text graph definition must be taken from opencv_extra:
1111
# https://github.com/opencv/opencv_extra/tree/master/testdata/dnn/ssd_mobilenet_v1_coco.pbtxt
1212
import numpy as np
@@ -40,7 +40,7 @@
4040
parser.add_argument("--num_classes", default=20, type=int,
4141
help="Number of classes. It's 20 for Caffe model from "
4242
"https://github.com/chuanqi305/MobileNet-SSD/ and 90 for "
43-
"TensorFlow model from https://github.com/tensorflow/models/tree/master/research/object_detection")
43+
"TensorFlow model from http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_11_06_2017.tar.gz")
4444
parser.add_argument("--thr", default=0.2, type=float, help="confidence threshold to filter out weak detections")
4545
args = parser.parse_args()
4646

0 commit comments

Comments
 (0)