如何在浏览器上执行TensorFlow实时对象检测模型?



我是一个机器学习初学者,在我的第一个ML项目上工作,我做了一个程序,检测人们是否实时戴着头盔(使用TensorFlow和OpenCV),现在当我执行它时,Python使用我的相机显示一个新窗口,但现在我想在浏览器上执行它,只是在本地,我不想部署它

这是执行和显示程序的代码部分:

while True: 
ret, frame = cap.read()
image_np = np.array(frame)

input_tensor = tf.convert_to_tensor(np.expand_dims(image_np, 0), dtype=tf.float32)
detections = detect_fn(input_tensor)

num_detections = int(detections.pop('num_detections'))
detections = {key: value[0, :num_detections].numpy()
for key, value in detections.items()}
detections['num_detections'] = num_detections
# detection_classes should be ints.
detections['detection_classes'] = detections['detection_classes'].astype(np.int64)
label_id_offset = 1
image_np_with_detections = image_np.copy()
viz_utils.visualize_boxes_and_labels_on_image_array(
image_np_with_detections,
detections['detection_boxes'],
detections['detection_classes']+label_id_offset,
detections['detection_scores'],
category_index,
use_normalized_coordinates=True,
max_boxes_to_draw=5,
min_score_thresh=.5,
agnostic_mode=False)
cv2.imshow('object detection',  cv2.resize(image_np_with_detections, (800, 600)))

if cv2.waitKey(1) & 0xFF == ord('q'):
cap.release()
break

我不知道如何从这里开始,它在我的电脑上工作得很好,但我不知道如何切换到浏览器,我不知道从哪里开始,如果可能的话,我会感激你的帮助

试试GoogleColabKaggle,它们都有你需要的。

尝试使用Flask,你可以在浏览器上执行它。

相关内容

  • 没有找到相关文章

最新更新