Skip to content
Snippets Groups Projects
Commit f49ab066 authored by Low, Low Jian He (PG/R - Comp Sci & Elec Eng)'s avatar Low, Low Jian He (PG/R - Comp Sci & Elec Eng)
Browse files

change for BSLcorpus

parent 2f528f2a
No related branches found
No related tags found
No related merge requests found
...@@ -154,8 +154,6 @@ def main(args, model, renderer, device): ...@@ -154,8 +154,6 @@ def main(args, model, renderer, device):
pred_bboxes = np.array([list(map(int, args.custom_bbox.split(',')))]) pred_bboxes = np.array([list(map(int, args.custom_bbox.split(',')))])
print(f"Using custom bbox, {pred_bboxes[0]}") print(f"Using custom bbox, {pred_bboxes[0]}")
show_bbox(img_cv2, pred_bboxes)
# Force confidence to be 0.99 that human is present # Force confidence to be 0.99 that human is present
pred_scores = np.array([0.99]) pred_scores = np.array([0.99])
...@@ -231,7 +229,7 @@ def main(args, model, renderer, device): ...@@ -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)}" 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 #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 = Path(temp_dir) / f"{File_Name}.lmdb"
Temp_Database.mkdir(parents=True, exist_ok=True) Temp_Database.mkdir(parents=True, exist_ok=True)
...@@ -303,8 +301,8 @@ if __name__ == '__main__': ...@@ -303,8 +301,8 @@ if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
args.bbox = 'False' args.bbox = 'False'
args.custom_bbox = '350,100,1600,1070' args.custom_bbox = '70,0,550,360'
# args.out_folder = os.path.join(args.out_folder, os.path.basename(os.path.parent(args.vid))) 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}") 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 model, model_cfg = load_hamer(args.checkpoint, load_mesh=False) # False sets model to not produce vertice on inference
......
...@@ -353,11 +353,12 @@ if __name__ == '__main__': ...@@ -353,11 +353,12 @@ if __name__ == '__main__':
Filename = 'BUSINESS_GROUP_AND_SET-UP_SPIN-OFF_PARTNER-RELATIONSHIP' Filename = 'BUSINESS_GROUP_AND_SET-UP_SPIN-OFF_PARTNER-RELATIONSHIP'
args.vid = f'/mnt/fast/nobackup/users/jl02958/Project/Rachel_Sentences/{Filename}.mp4' 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/Rachel_HaMeR/Sentences/{Filename}.lmdb' 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/{Filename}' args.out_folder = f'/vol/research/signVision/Projects/BSLboundaries/Test/BF17n'
args.frame_range = '30,400'
args.bbox = True args.bbox = False
args.custom_bbox = '70,0,550,360'
args.save_mesh = False args.save_mesh = False
# Download and load checkpoints # Download and load checkpoints
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment