Skip to content

Commit e843880

Browse files
authored
Update SSD_demo.py
1 parent f120225 commit e843880

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

ObjectDetections/SSD/SSD_demo.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
from visualization import plt_bboxes
1313

1414

15-
1615
ssd_net = SSD()
1716
classes, scores, bboxes = ssd_net.detections()
1817
images = ssd_net.images()
@@ -27,14 +26,8 @@
2726
img = cv2.imread('./demo/dog.jpg')
2827
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
2928
img_prepocessed = preprocess_image(img)
30-
3129
rclasses, rscores, rbboxes = sess.run([classes, scores, bboxes],
3230
feed_dict={images: img_prepocessed})
33-
34-
3531
rclasses, rscores, rbboxes = process_bboxes(rclasses, rscores, rbboxes)
3632

3733
plt_bboxes(img, rclasses, rscores, rbboxes)
38-
39-
40-

0 commit comments

Comments
 (0)