diff --git a/Extract_HaMeR_Single.py b/Extract_HaMeR_Single.py index fb9e5bea54f312dc7118800fe5d3a8ad488ae23d..69afc85883c7a6512acf74f3b68fae77d4d9273d 100644 --- a/Extract_HaMeR_Single.py +++ b/Extract_HaMeR_Single.py @@ -198,7 +198,7 @@ def main(args, model, renderer, device): img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removesuffix('.png').removeprefix('Frame'))) img_cv2 = cv2.imread(str(img_paths[0])) - if args.bbox == True: + if args.bbox == 'True': # Detect humans in image det_out = detector(img_cv2) det_instances = det_out['instances'] @@ -328,7 +328,7 @@ if __name__ == '__main__': parser.add_argument('--rescale_factor', type=float, default=2.0, help='Factor for padding the bbox') parser.add_argument('--body_detector', type=str, default='vitdet', choices=['vitdet', 'regnety'], help='Using regnety improves runtime and reduces memory') parser.add_argument('--file_type', nargs='+', default=['*.jpg', '*.png'], help='List of file extensions to consider') - parser.add_argument('--bbox', type=bool, default=True, help= 'If set, use provided bbox from ViT') + parser.add_argument('--bbox', type=str, default='True', help= 'If set, use provided bbox from ViT') parser.add_argument('--custom_bbox', type=str, default='', help='Custom bbox in the format x1,y1,x2,y2') parser.add_argument('--MANO_Output', type=bool, default=False, help= 'If set, generate output images') diff --git a/Inject_Json.py b/Inject_Json.py index 20c68421df9056f81f6c1fccef92ac1bb4a010c5..0f5af21b4dc9502a0f1b4610a99eae92d8c6d0bc 100644 --- a/Inject_Json.py +++ b/Inject_Json.py @@ -314,10 +314,11 @@ if __name__ == '__main__': parser.add_argument('--frame_range', type=str, default='0,-1', help='Frame range for input') args = parser.parse_args() - args.vid = '/vol/vssp/datasets/mixedmode/mein-dgs-korpus/CROPPED_VIDEOS/1176549/1176549_1a1.cropped.mp4' - args.json_folder = '/vol/research/signVision/Projects/BSLboundaries/MeinDGS_HaMeR_Feature/1176549/1176549_1a1.cropped.tar.xz' + args.vid = '/vol/vssp/datasets/mixedmode/mein-dgs-korpus/RawData/1176549/1176549_1a1.mp4' + args.json_folder = '/vol/research/signVision/Projects/BSLboundaries/MeinDGS_HaMeR_Feature/1176549/1176549_1a1.tar.xz' args.out_folder = '/vol/research/signVision/Projects/BSLboundaries/Test' args.frame_range = '0,150' + args.custom_bbox = '100,0,530,360' args.bbox = False # Download and load checkpoints