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

save

Browse Source
This commit is contained in:
luni
2023-03-04 12:36:00 +00:00
parent 165adf7953
commit 7c92e03b45
28 changed files with 1216 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dataset.py
+8 -8
View File
@@ -381,18 +381,18 @@ if __name__ == '__main__':
# 'uv_shaped_body': (3, 512, 256,),
}
dataset = factory.get_dataset(folder='/home/workspace/local-data/train_skirt',
batch_size=1,
dataset = factory.get_dataset(folder='/home/workspace/local-data/cape_processed/train/male',
batch_size=10,
signature=signature_dynamic,
resolutions=[512],
do_repeat=False,
use_one_hot=False,
is_cape=False,
frames_file='frames_dress_30k.txt')
static_dataset = StaticDataset(folder='/home/workspace/local-data/train_skirt',
is_cape=True)
# frames_file='frames_dress_30k.txt'
static_dataset = StaticDataset(folder='/home/workspace/local-data/cape_processed/train/male',
signature=signature_static,
resolutions=[512],
is_cape=False)
is_cape=True)
# for i, batch in enumerate(dataset):
@@ -404,7 +404,7 @@ if __name__ == '__main__':
for i, batch in enumerate(dataset):
print(i)
batch = static_dataset.expand_batch(batch_to_dict(batch, signature_dynamic))
ex.imshow(batch['uv_curvature'][0].numpy().argmax(-1) / 6.0).show()
# ex.imshow(batch['uv_curvature'][0].numpy().argmax(-1) / 6.0).show()
sum_uv += batch['uv_unposed'].numpy().sum(axis=0)
sum_mask += batch['uv_mask'].numpy().sum(axis=0)
@@ -425,7 +425,7 @@ if __name__ == '__main__':
k = F[:,2]
)]).show()
np.save('custom_data/train_dress_30k_no_shape_avg_uv.npy', avg_garment)
np.save('custom_data/train_cape_m_no_shape_avg_uv.npy', avg_garment)
print()