Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from glob import glob
- import os
- from shutil import copy
- # sample keypoint filename: HICO_test2015_00009750_keypoints.json
- test_keypoints = '/media/mona/easystore/human_pose_estimation/data/smplify-x/djrn_test_data/keypoints/'
- train_keypoints = '/media/mona/easystore/human_pose_estimation/data/smplify-x/djrn_train_data/keypoints/'
- # sample image name: HICO_test2015_00000330.jpg
- ride_bike_test = '/home/mona/research/code/ConsNet/data/ride_bike_djrn/test/ride_bicycle/'
- ride_bike_train = '/home/mona/research/code/ConsNet/data/ride_bike_djrn/train/ride_bicycle/'
- test_keypoints_ride_bike = 'ride_bike_keypoints/test/keypoints'
- train_keypoints_ride_bike = 'ride_bike_keypoints/train/keypoints'
- if not os.path.exists(test_keypoints_ride_bike):
- os.makedirs(test_keypoints_ride_bike)
- if not os.path.exists(train_keypoints_ride_bike):
- os.makedirs(train_keypoints_ride_bike)
- for filename in glob(ride_bike_test + "*.jpg"):
- for keypoint_f in glob(test_keypoints + "*.json"):
- if os.path.basename(filename)[:-4] in keypoint_f:
- copy(keypoint_f, test_keypoints_ride_bike)
- #import pdb
- for filename in glob(ride_bike_train + "*.jpg"):
- for keypoint_f in glob(train_keypoints + "*.json"):
- if os.path.basename(filename)[:-4] in keypoint_f:
- #pdb.set_trace()
- copy(keypoint_f, train_keypoints_ride_bike)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement