一凡~
2021-03-24 00:42
采纳率: 100%
浏览 45

机器学习python的imageal包进行物体检测提取时总是显卡内存不足

提取出如下图片人脸

from imageai.Detection import ObjectDetection

import os

execution_path = os.getcwd()
detector = ObjectDetection()
detector.setModelTypeAsTinyYOLOv3()
detector.setModelPath( os.path.join(execution_path , "yolo-tiny.h5"))
detector.loadModel()
detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image2.jpg"), output_image_path=os.path.join(execution_path , "image2new.jpg"), minimum_percentage_probability=30)

for eachObject in detections:
    print(eachObject["name"] , " : ", eachObject["percentage_probability"], " : ", eachObject["box_points"] )
    print("--------------------------------")

错误如下

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

2条回答 默认 最新

  • coagenth 2021-03-24 10:21
    已采纳

    这个只是warnings吧,我的机器没有AVX指令集,安装github上无GPU的tensorflow版本,运行你的代码正常啊,不过我用的模型是yolo.h5

    记住要把路径指向到h5文件。

    点赞 打赏 评论
  • 一凡~ 2021-03-24 00:43

    求大佬

    点赞 打赏 评论

相关推荐 更多相似问题