@@ -27,8 +27,11 @@ class FaceAnalysis:
self.model_dir = ensure_available('models', name, root=root)
onnx_files = glob.glob(osp.join(self.model_dir, '*.onnx'))
onnx_files = sorted(onnx_files)
- for onnx_file in onnx_files:
- model = model_zoo.get_model(onnx_file, **kwargs)
+ om_files = glob.glob(osp.join(self.model_dir, '*.om'))
+ om_files = sorted(om_files)
+ for onnx_file, om_file in zip (onnx_files, om_files):
+ print(f"onnx_file: {onnx_file}")
+ model = model_zoo.get_model(onnx_file, om_file, **kwargs)
if model is None:
print('model not recognized:', onnx_file)
elif allowed_modules is not None and model.taskname not in allowed_modules:
@@ -81,11 +84,11 @@ class FaceAnalysis:
dimg = img.copy()
for i in range(len(faces)):
face = faces[i]
- box = face.bbox.astype(np.int)
+ box = face.bbox.astype(int)
color = (0, 0, 255)
cv2.rectangle(dimg, (box[0], box[1]), (box[2], box[3]), color, 2)
if face.kps is not None:
- kps = face.kps.astype(np.int)
+ kps = face.kps.astype(int)
#print(landmark.shape)
for l in range(kps.shape[0]):
color = (0, 0, 255)
