Skip to content

Commit 6e9e185

Browse files
committed
blaze_detect_live: update for working vitis-ai models. display profile log when in debug mode.
1 parent 2fdc403 commit 6e9e185

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

blaze_detect_live.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -201,9 +201,8 @@ def get_video_dev_by_name(src):
201201
{ "blaze": "face", "pipeline": "tfl_face_v0_10_sparse", "model1": "blaze_tflite/models/face_detection_full_range_sparse.tflite", "model2": "blaze_tflite/models/face_landmark.tflite" },
202202
{ "blaze": "face", "pipeline": "pyt_face_v0_07_front" , "model1": "blaze_pytorch/models/blazeface.pth", "model2": "blaze_pytorch/models/blazeface_landmark.pth" },
203203
{ "blaze": "face", "pipeline": "pyt_face_v0_07_back" , "model1": "blaze_pytorch/models/blazefaceback.pth", "model2": "blaze_pytorch/models/blazeface_landmark.pth" },
204-
{ "blaze": "face", "pipeline": "vai_face_v0_07_hybrid", "model1": "blaze_tflite/models/face_detection_front_v0_07.tflite", "model2": "blaze_vitisai/models/BlazeFaceLandmark/"+dpu_arch+"/BlazeFaceLandmark.xmodel" },
205-
{ "blaze": "face", "pipeline": "vai_face_v0_07_front" , "model1": "blaze_vitisai/models/BlazeFace/"+dpu_arch+"/BlazeFace.xmodel","model2": "blaze_vitisai/models/BlazeFaceLandmark/"+dpu_arch+"/BlazeFaceLandmark.xmodel" },
206-
{ "blaze": "face", "pipeline": "vai_face_v0_07_back" , "model1": "blaze_vitisai/models/BlazeFaceBack/"+dpu_arch+"/BlazeFaceBack.xmodel","model2": "blaze_vitisai/models/BlazeFaceLandmark/"+dpu_arch+"/BlazeFaceLandmark.xmodel" },
204+
# { "blaze": "face", "pipeline": "vai_face_v0_07_front" , "model1": "blaze_vitisai/models/BlazeFace/"+dpu_arch+"/BlazeFace.xmodel","model2": "blaze_vitisai/models/BlazeFaceLandmark/"+dpu_arch+"/BlazeFaceLandmark.xmodel" },
205+
# { "blaze": "face", "pipeline": "vai_face_v0_07_back" , "model1": "blaze_vitisai/models/BlazeFaceBack/"+dpu_arch+"/BlazeFaceBack.xmodel","model2": "blaze_vitisai/models/BlazeFaceLandmark/"+dpu_arch+"/BlazeFaceLandmark.xmodel" },
207206
{ "blaze": "face", "pipeline": "hai_face_v0_10_short" , "model1": "blaze_hailo/models/face_detection_short_range.hef", "model2": "blaze_hailo/models/face_landmark.hef" },
208207
{ "blaze": "face", "pipeline": "hai_face_v0_10_full" , "model1": "blaze_hailo/models/face_detection_full_range.hef", "model2": "blaze_hailo/models/face_landmark.hef" },
209208
{ "blaze": "pose", "pipeline": "tfl_pose_v0_10_lite" , "model1": "blaze_tflite/models/pose_detection.tflite", "model2": "blaze_tflite/models/pose_landmark_lite.tflite" },
@@ -656,7 +655,8 @@ def ignore(x):
656655
str(prof_annotate[pipeline_id])+","+\
657656
str(prof_total[pipeline_id])+","+\
658657
str(prof_fps[pipeline_id])+"\n"
659-
#print("[LOG] ",csv_str)
658+
if args.debug:
659+
print("[LOG] ",csv_str)
660660
f_profile_csv.write(csv_str)
661661

662662

0 commit comments

Comments
 (0)