diff --git a/Extract_Uncompressed_HaMeR.py b/Extract_Uncompressed_HaMeR.py index d40589ec530d2b0d23c615f86762664130c409d1..1b12d0b0eee6030160b8d38d3e29d995292debb5 100644 --- a/Extract_Uncompressed_HaMeR.py +++ b/Extract_Uncompressed_HaMeR.py @@ -196,6 +196,7 @@ def main(args, model, renderer, device): img_cv2 = cv2.imread(str(img_paths[0])) det_out = detector(img_cv2) det_instances = det_out['instances'] + print('Completed ViTDet for Human Bounding Box Aquisition') print_gpu_usage() #Clearing memory del detector @@ -214,6 +215,7 @@ def main(args, model, renderer, device): # keypoint detector cpm = ViTPoseModel(device) + print('Loading ViTPose Model') print_gpu_usage() else: # Hardcoded bbox. This assumes person is in the center and that there is always one person in the image