| dataset: | |
| name: VITON-HD (Zalando HD Resized) | |
| root: d:\Projects\AI-Virtual-TryOn\data\zalando-hd-resized | |
| train_samples: 10482 | |
| val_samples: 1165 | |
| test_samples: 2032 | |
| total_samples: 13679 | |
| preprocessing: | |
| image_size: !!python/tuple | |
| - 512 | |
| - 384 | |
| normalization: | |
| mean: | |
| - 0.5 | |
| - 0.5 | |
| - 0.5 | |
| std: | |
| - 0.5 | |
| - 0.5 | |
| - 0.5 | |
| augmentation: true | |
| augmentation_techniques: | |
| - ColorJitter | |
| - GaussianBlur | |
| - MedianBlur | |
| dataloader: | |
| batch_size: 8 | |
| num_workers: 4 | |
| shuffle: true | |
| pin_memory: true | |
| pose: | |
| num_keypoints: 25 | |
| format: OpenPose | |
| keypoint_format: '[x, y, confidence]' | |
| segmentation: | |
| num_classes: 12 | |
| classes: | |
| - 0 | |
| - 9 | |
| - 28 | |
| - 38 | |
| - 55 | |
| - 59 | |
| - 75 | |
| - 125 | |
| - 139 | |
| - 178 | |
| - 193 | |
| - 209 | |
| class_names: | |
| 0: Background | |
| 1: Hat | |
| 2: Hair | |
| 3: Glove | |
| 4: Sunglasses | |
| 5: Upper-clothes | |
| 6: Dress | |
| 7: Coat | |
| 8: Socks | |
| 9: Pants | |
| 10: Jumpsuits | |
| 11: Scarf | |
| 12: Skirt | |
| 13: Face | |
| 14: Left-arm | |
| 15: Right-arm | |
| 16: Left-leg | |
| 17: Right-leg | |
| 18: Left-shoe | |
| 19: Right-shoe | |