diff --git a/Extract_HaMeR_Single.py b/Extract_HaMeR_Single.py index 688aed9c10d671662d096aff54f086f8372049ed..94544e590330420e8d5f5116f56d12acef5e73f6 100644 --- a/Extract_HaMeR_Single.py +++ b/Extract_HaMeR_Single.py @@ -10,7 +10,7 @@ import matplotlib.pyplot as plt import matplotlib.patches as patches import gc import shutil -import lmdb, pickle +# import lmdb, pickle from hamer.configs import CACHE_DIR_HAMER from hamer.models import HAMER, download_models, load_hamer, DEFAULT_CHECKPOINT @@ -217,44 +217,44 @@ def main(args, model, renderer, device): # Pickling Protocol set to 4 protocol = 4 - with lmdb.open(path=str(Temp_Database), map_size=n_bytes) as env: - with env.begin(write=True) as txn: - for index,feat in enumerate(global_orient_list): - txn.put( - key=f"{File_Name}_GOrient_{index}".encode("ascii"), - value=pickle.dumps(feat, protocol=protocol), - dupdata=False, - ) - - with env.begin(write=True) as txn: - for index,feat in enumerate(hand_pose_list): - txn.put( - key=f"{File_Name}_HPose_{index}".encode("ascii"), - value=pickle.dumps(feat, protocol=protocol), - dupdata=False, - ) - - with env.begin(write=True) as txn: - txn.put( - key=("pred_cam").encode("ascii"), - value=pickle.dumps(pred_cam_list, protocol=protocol), - dupdata=False, - ) + # with lmdb.open(path=str(Temp_Database), map_size=n_bytes) as env: + # with env.begin(write=True) as txn: + # for index,feat in enumerate(global_orient_list): + # txn.put( + # key=f"{File_Name}_GOrient_{index}".encode("ascii"), + # value=pickle.dumps(feat, protocol=protocol), + # dupdata=False, + # ) + + # with env.begin(write=True) as txn: + # for index,feat in enumerate(hand_pose_list): + # txn.put( + # key=f"{File_Name}_HPose_{index}".encode("ascii"), + # value=pickle.dumps(feat, protocol=protocol), + # dupdata=False, + # ) + + # with env.begin(write=True) as txn: + # txn.put( + # key=("pred_cam").encode("ascii"), + # value=pickle.dumps(pred_cam_list, protocol=protocol), + # dupdata=False, + # ) - with env.begin(write=True) as txn: - txn.put( - key=("betas").encode("ascii"), - value=pickle.dumps(betas_list, protocol=protocol), - dupdata=False, - ) - - with env.begin(write=True) as txn: - txn.put( - key=("details").encode("ascii"), - value=pickle.dumps({"num_features": list_length, "fps": fps, "focal_length": focal_length}, protocol=protocol), - dupdata=False, - ) - shutil.move(f"{Temp_Database}", args.out_folder) + # with env.begin(write=True) as txn: + # txn.put( + # key=("betas").encode("ascii"), + # value=pickle.dumps(betas_list, protocol=protocol), + # dupdata=False, + # ) + + # with env.begin(write=True) as txn: + # txn.put( + # key=("details").encode("ascii"), + # value=pickle.dumps({"num_features": list_length, "fps": fps, "focal_length": focal_length}, protocol=protocol), + # dupdata=False, + # ) + # shutil.move(f"{Temp_Database}", args.out_folder) print(f"Total time taken: {(time.time()-initial_start_time)/60:.2f} minutes") if __name__ == '__main__':