File size: 1,137 Bytes
b1d6c47 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 | 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
|