Leffa / leffa /conf /datasets /deepfashion_test.yaml
franciszzj's picture
init code
b213d84
raw
history blame
1.9 kB
deepfashion_test:
dataset:
_target_: media_dataloader.api.EnrichingDataset
datasource:
_target_: media_dataloader.api.LazyHiveDataSource
namespace: mgenai
table: deepfashion_pose_transfer
partition_filter_predicate_list: ["ds = '2024-08-15' AND set_name = 'val'"]
enrichments:
- _target_: media_dataloader.api.media_lookups.ManifoldLookups
lookup_handle_to_media_columns:
to_img_manifold_path: "image"
from_img_manifold_path: "cloth"
to_img_iuv_manifold_path: "image_densepose"
from_img_iuv_manifold_path: "cloth_densepose"
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: image_densepose
blob_field: image_densepose
- _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
image_field: cloth_densepose
blob_field: cloth_densepose
- _target_: leffa.datasets.transform.VtonTransform
height: ${constants.height}
width: ${constants.width}
is_train: false
dataset: deepfashion
aug_garment_ratio: 0.0
get_garment_from_person_ratio: 0.0
aug_mask_ratio: 0.0
dataloader:
_target_: media_dataloader.api.StatefulDataLoader
dataset: ${datasets.deepfashion_test.dataset}
batch_size: ${constants.batch_size}
num_workers: 0
prefetch_factor: null
pin_memory: true
persistent_workers: false
multiprocessing_context: null