'update-mnn'

@@ -3,9 +3,10 @@ import numpy as np
 import MNN
 import os
 
+
 def image_infer_mnn(mnn_model_path, image_path, class_list):
     image = cv2.imread(image_path)
-    input_image = cv2.resize(image,(128,128))
+    input_image = cv2.resize(image, (128, 128))
     input_data = input_image.astype(np.float32).transpose((2, 0, 1)) / 255
     interpreter = MNN.Interpreter(mnn_model_path)
     session = interpreter.createSession()
@@ -17,7 +18,7 @@ def image_infer_mnn(mnn_model_path, image_path, class_list):
     output_data = infer_result.getData()
     out = output_data.index(max(output_data))
 
-    cv2.putText(image,class_list[int(out)],(50, 50),cv2.FONT_HERSHEY_SIMPLEX,2,(0,0,255))
+    cv2.putText(image, class_list[int(out)], (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255))
     return image
 
 
@@ -74,10 +75,10 @@ if __name__ == '__main__':
     image_path = 'test_image/mask_2997.jpg'
     mnn_model_path = 'cls_abnormal_face_mnn_1.0.0_v0.0.1.mnn'
     # image
-    # for i in os.listdir('test_image'):
-    #     image=image_infer_mnn(mnn_model_path,os.path.join('test_image',i),class_list)
-    #     cv2.imshow('image',image)
-    #     cv2.waitKey(0)
+    for i in os.listdir('test_image'):
+        image=image_infer_mnn(mnn_model_path,os.path.join('test_image',i),class_list)
+        cv2.imshow('image',image)
+        cv2.waitKey(0)
 
     # camera
-    camera_infer_mnn(mnn_model_path, 0)
+    # camera_infer_mnn(mnn_model_path, 0)