From f49ab0666005a63fc08ced0594245ef2ed4119e6 Mon Sep 17 00:00:00 2001
From: JianHe0628 <jl02958@surrey.ac.uk>
Date: Tue, 10 Dec 2024 19:45:28 +0000
Subject: [PATCH] change for BSLcorpus
---
Extract_HaMeR_Single.py | 8 +++-----
Inject_Json.py | 11 ++++++-----
2 files changed, 9 insertions(+), 10 deletions(-)
diff --git a/Extract_HaMeR_Single.py b/Extract_HaMeR_Single.py
index e3fca51..a10f7c8 100644
--- a/Extract_HaMeR_Single.py
+++ b/Extract_HaMeR_Single.py
@@ -154,8 +154,6 @@ def main(args, model, renderer, device):
pred_bboxes = np.array([list(map(int, args.custom_bbox.split(',')))])
print(f"Using custom bbox, {pred_bboxes[0]}")
- show_bbox(img_cv2, pred_bboxes)
-
# Force confidence to be 0.99 that human is present
pred_scores = np.array([0.99])
@@ -231,7 +229,7 @@ def main(args, model, renderer, device):
assert len(pred_cam_list) == len(img_paths), f"Length of output features and input images are not equal. features: {len(pred_cam_list)}, img_paths: {len(img_paths)}"
#Create Temp Out Folder for lmdb
- File_Name = os.path.basename(args.vid).removesuffix(".mp4") if args.vid != '' else os.path.basename(args.img_folder)
+ File_Name = os.path.basename(args.vid).removesuffix(".mp4").removesuffix(".mov") if args.vid != '' else os.path.basename(args.img_folder)
Temp_Database = Path(temp_dir) / f"{File_Name}.lmdb"
Temp_Database.mkdir(parents=True, exist_ok=True)
@@ -303,8 +301,8 @@ if __name__ == '__main__':
args = parser.parse_args()
args.bbox = 'False'
- args.custom_bbox = '350,100,1600,1070'
- # args.out_folder = os.path.join(args.out_folder, os.path.basename(os.path.parent(args.vid)))
+ args.custom_bbox = '70,0,550,360'
+ args.out_folder = os.path.join(args.out_folder, os.path.dirname(args.vid).removeprefix('/vol/vssp/LF_datasets/multiview/bslcp/BSLCP/'))
print(f"Output folder: {args.out_folder}")
model, model_cfg = load_hamer(args.checkpoint, load_mesh=False) # False sets model to not produce vertice on inference
diff --git a/Inject_Json.py b/Inject_Json.py
index 18cbd8b..27a55aa 100644
--- a/Inject_Json.py
+++ b/Inject_Json.py
@@ -353,11 +353,12 @@ if __name__ == '__main__':
Filename = 'BUSINESS_GROUP_AND_SET-UP_SPIN-OFF_PARTNER-RELATIONSHIP'
- args.vid = f'/mnt/fast/nobackup/users/jl02958/Project/Rachel_Sentences/{Filename}.mp4'
- args.lmdb_input = f'/mnt/fast/nobackup/scratch4weeks/jl02958/Rachel_HaMeR/Sentences/{Filename}.lmdb'
- args.out_folder = f'/vol/research/signVision/Projects/BSLboundaries/Test/{Filename}'
-
- args.bbox = True
+ args.vid = f'/vol/vssp/LF_datasets/multiview/bslcp/BSLCP/Videos/Narratives/Belfast/17+18/BF17n.mov'
+ args.lmdb_input = f'/mnt/fast/nobackup/scratch4weeks/jl02958/BSLCorpus/Videos/Narratives/Belfast/17+18/BF17n.lmdb'
+ args.out_folder = f'/vol/research/signVision/Projects/BSLboundaries/Test/BF17n'
+ args.frame_range = '30,400'
+ args.bbox = False
+ args.custom_bbox = '70,0,550,360'
args.save_mesh = False
# Download and load checkpoints
--
GitLab