File size: 1,898 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
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