From 609d853fc1371cfe7996e55f2a21777cc0daf6bb Mon Sep 17 00:00:00 2001
From: JianHe0628 <jl02958@surrey.ac.uk>
Date: Tue, 19 Nov 2024 17:54:07 +0000
Subject: [PATCH] Adding for informative prints for GPU usage
---
Extract_Uncompressed_HaMeR.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/Extract_Uncompressed_HaMeR.py b/Extract_Uncompressed_HaMeR.py
index d40589e..1b12d0b 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
--
GitLab