Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,553 Bytes
b213d84 |
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 |
dress_code_test:
dataset:
_target_: media_dataloader.api.EnrichingDataset
datasource:
_target_: media_dataloader.api.LazyHiveDataSource
namespace: ad_metrics
table: vton_public_dataset_dress_code_test_paired_v2
# table: vton_public_dataset_dress_code_test_unpaired_v2
# table: vton_public_dataset_dress_code_test_upper_body_paired_v2
# table: vton_public_dataset_dress_code_test_upper_body_unpaired_v2
partition_filter_predicate_list: ["ds = '2024-09-14'"]
# table: vton_public_dataset_dress_code_test_lower_body_paired_v2
# table: vton_public_dataset_dress_code_test_lower_body_unpaired_v2
# table: vton_public_dataset_dress_code_test_dresses_paired_v2
# table: vton_public_dataset_dress_code_test_dresses_unpaired_v2
# partition_filter_predicate_list: ["ds = '2024-09-16'"]
enrichments:
- _target_: media_dataloader.api.media_lookups.ManifoldLookups
lookup_handle_to_media_columns:
image_manifold_path: "image"
cloth_manifold_path: "cloth"
agnostic_mask_manifold_path: "agnostic_mask"
dense_manifold_path: "image_densepose"
label_map_manifold_path: "image_parse"
collate_fn:
- _target_: media_dataloader.api.Collate
- _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
image_field: image
blob_field: image
- _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
image_field: cloth
blob_field: cloth
- _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
image_field: agnostic_mask
blob_field: agnostic_mask
- _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
image_field: image_densepose
blob_field: image_densepose
- _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
image_field: image_parse
blob_field: image_parse
- _target_: leffa.datasets.transform.VtonTransform
height: ${constants.height}
width: ${constants.width}
is_train: false
dataset: dress_code
dataloader:
_target_: media_dataloader.api.StatefulDataLoader
dataset: ${datasets.dress_code_test.dataset}
batch_size: ${constants.batch_size}
num_workers: 0
prefetch_factor: null
pin_memory: true
persistent_workers: false
multiprocessing_context: null
|