luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

commit all

Browse Source
This commit is contained in:
luni
2023-01-25 12:25:59 +00:00
parent 1bf224841f
commit 7470ca9fb6
139 changed files with 82069 additions and 811 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cluster_poses.py
+4 -4
View File
@@ -50,13 +50,13 @@ def sample_from_cluster(file, n_sample, output_file):
if __name__ == '__main__':
folder = '/home/workspace/data/dataset/test_ready'
folder = '/home/workspace/data/dataset/test_skirt_ready_2'
is_skirt = False
is_skirt = True
F = np.load("custom_utils/smpl_faces" + ("_skirt" if is_skirt == 1 else "") + ".npy")
# generate_clusters(folder, 20)
sample_from_cluster('custom_data/test_ready_pose_cluster_20.csv', 10, '/home/workspace/data/dataset/test_ready/frames_pose_cluster_20_sample_10.txt')
generate_clusters(folder, 20)
# sample_from_cluster('custom_data/test_skirt_ready_pose_cluster_20.csv', 10, '/home/workspace/data/dataset/test_skirt_ready/frames_pose_cluster_20_sample_10.txt')
# samples = df.groupby('label', group_keys=False).apply(lambda x: x.sample())