- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
Hello,
when I run this code
import cv2
import time
print (cv2.__version__)
start = time.time()
# Pretrained classes in the model
classNames = {0: 'background',
1: 'person', 2: 'bicycle', 3: 'car', 4: 'motorcycle', 5: 'airplane', 6: 'bus',
7: 'train', 8: 'truck', 9: 'boat', 10: 'traffic light', 11: 'fire hydrant',
13: 'stop sign', 14: 'parking meter', 15: 'bench', 16: 'bird', 17: 'cat',
18: 'dog', 19: 'horse', 20: 'sheep', 21: 'cow', 22: 'elephant', 23: 'bear',
24: 'zebra', 25: 'giraffe', 27: 'backpack', 28: 'umbrella', 31: 'handbag',
32: 'tie', 33: 'suitcase', 34: 'frisbee', 35: 'skis', 36: 'snowboard',
37: 'sports ball', 38: 'kite', 39: 'baseball bat', 40: 'baseball glove',
41: 'skateboard', 42: 'surfboard', 43: 'tennis racket', 44: 'bottle',
46: 'wine glass', 47: 'cup', 48: 'fork', 49: 'knife', 50: 'spoon',
51: 'bowl', 52: 'banana', 53: 'apple', 54: 'sandwich', 55: 'orange',
56: 'broccoli', 57: 'carrot', 58: 'hot dog', 59: 'pizza', 60: 'donut',
61: 'cake', 62: 'chair', 63: 'couch', 64: 'potted plant', 65: 'bed',
67: 'dining table', 70: 'toilet', 72: 'tv', 73: 'laptop', 74: 'mouse',
75: 'remote', 76: 'keyboard', 77: 'cell phone', 78: 'microwave', 79: 'oven',
80: 'toaster', 81: 'sink', 82: 'refrigerator', 84: 'book', 85: 'clock',
86: 'vase', 87: 'scissors', 88: 'teddy bear', 89: 'hair drier', 90: 'toothbrush'}
def id_class_name(class_id, classes):
for key, value in classes.items():
if class_id == key:
return value
# Loading model
model = cv2.dnn.readNetFromTensorflow('frozen_inference_graph.pb',
'ssd_mobilenet_v2_coco_2018_03_29.pbtxt')
model.setPreferableTarget(cv2.dnn.DNN_TARGET_MYRIAD)
image = cv2.imread("image.jpeg")
image_height, image_width, _ = image.shape
model.setInput(cv2.dnn.blobFromImage(image, size=(100, 100), swapRB=True))
output = model.forward()
# print(output[0,0,:,:].shape)
for detection in output[0, 0, :, :]:
confidence = detection[2]
if confidence > .5:
class_id = detection[1]
class_name=id_class_name(class_id,classNames)
print(str(str(class_id) + " " + str(detection[2]) + " " + class_name))
box_x = detection[3] * image_width
box_y = detection[4] * image_height
box_width = detection[5] * image_width
box_height = detection[6] * image_height
cv2.rectangle(image, (int(box_x), int(box_y)), (int(box_width), int(box_height)), (23, 230, 210), thickness=1)
cv2.putText(image,class_name ,(int(box_x), int(box_y+.05*image_height)),cv2.FONT_HERSHEY_SIMPLEX,(.005*image_width),(0, 0, 255))
cv2.imshow('image', image)
# cv2.imwrite("image_box_text.jpg",image)
end = time.time()
print("[INFO] SSD took {:.6f} seconds".format(end - start))
cv2.waitKey(0)
cv2.destroyAllWindows()
the NCS2 under Windows 10 becomes this device:
normal it's this one:
Also, the stick is very slow on another PC. My suspicion is that the CPU is doing the work and not the stick. As it turns into a different device.
Does anyone have any idea what can be done?
Thanks
Peter
Link Copied
2 Replies
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
UPDATE:
The speed problem it's only by the first frame or image..it's this normal?
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
Hi Toppel, Peter
Which version of OpenVINO are you using?
Please confirm you followed the instructions to install NCS2 driver manually on windows and successfully executed the verification scripts.
There is a similar issue in the other thread which you can refer to.
Let us know in case of any other queries.
Best Regards,
Surya
Reply
Topic Options
- Subscribe to RSS Feed
- Mark Topic as New
- Mark Topic as Read
- Float this Topic for Current User
- Bookmark
- Subscribe
- Printer Friendly Page