実現したいこと
Python で insightface を使って顔認識をしたいのですが実行したときに ONNXRuntime に関してエラーが出ます
どこにバグがあるのか教えていただけますでしょうか。
前提
Python3系でPycharm上でinsightfaceを使用
発生している問題・エラーメッセージ
エラーメッセージ
[ WARN:0@2.087] global loadsave.cpp:244 cv::findDecoder imread_('input.jpg'): can't open/read file: check file path/integrity
C:\Users\user\PycharmProjects\insightFace\venv\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py:54: UserWarning: Specified provider 'CUDAExecutionProvider' is not in available provider names.Available providers: 'CPUExecutionProvider'
warnings.warn(
Applied providers: ['CPUExecutionProvider'], with options: {'CPUExecutionProvider': {}}
find model: C:\Users\user/.insightface\models\buffalo_l\1k3d68.onnx landmark_3d_68 ['None', 3, 192, 192] 0.0 1.0
Applied providers: ['CPUExecutionProvider'], with options: {'CPUExecutionProvider': {}}
find model: C:\Users\user/.insightface\models\buffalo_l\2d106det.onnx landmark_2d_106 ['None', 3, 192, 192] 0.0 1.0
Applied providers: ['CPUExecutionProvider'], with options: {'CPUExecutionProvider': {}}
find model: C:\Users\user/.insightface\models\buffalo_l\det_10g.onnx detection [1, 3, '?', '?'] 127.5 128.0
Applied providers: ['CPUExecutionProvider'], with options: {'CPUExecutionProvider': {}}
find model: C:\Users\user/.insightface\models\buffalo_l\genderage.onnx genderage ['None', 3, 96, 96] 0.0 1.0
Applied providers: ['CPUExecutionProvider'], with options: {'CPUExecutionProvider': {}}
find model: C:\Users\user/.insightface\models\buffalo_l\w600k_r50.onnx recognition ['None', 3, 112, 112] 127.5 127.5
set det-size: (640, 640)
Traceback (most recent call last):
File "C:\Users\user\PycharmProjects\insightFace\main.py", line 12, in <module>
faces = app.get(img)
File "C:\Users\user\PycharmProjects\insightFace\venv\lib\site-packages\insightface\app\face_analysis.py", line 59, in get
bboxes, kpss = self.det_model.detect(img,
File "C:\Users\user\PycharmProjects\insightFace\venv\lib\site-packages\insightface\model_zoo\retinaface.py", line 211, in detect
im_ratio = float(img.shape[0]) / img.shape[1]
AttributeError: 'NoneType' object has no attribute 'shape'
該当のソースコード
import numpy as np
import cv2
import insightface
from insightface.app import FaceAnalysis
image_file = "input.jpg"
img = cv2.imread(image_file)
app = insightface.app.FaceAnalysis()
app.prepare(ctx_id=1, det_size=(640, 640))
faces = app.get(img)
rimg = app.draw_on(img, faces)
cv2.imwrite("./output.jpg", rimg)
試したこと
app = insightface.app.FaceAnalysis()
app.prepare(ctx_id=1, det_size=(640, 640))
ここに問題があるのかなと思いました

回答1件
あなたの回答
tips
プレビュー