Spaces:
Running
Running
correct create_loader
Browse files
modules/dataset_loader.py
CHANGED
@@ -531,9 +531,9 @@ def collate_fn(batch):
|
|
531 |
|
532 |
|
533 |
def create_loader(new_size, transformation, annotations1, annotations2=None,
|
534 |
-
batch_size=4, crop_prob=0.
|
535 |
-
h_flip_prob=0.
|
536 |
-
seed=42, resize=True, keep_ratio=
|
537 |
"""
|
538 |
Create a DataLoader for BPMN datasets with optional transformations and concatenation of two datasets.
|
539 |
|
|
|
531 |
|
532 |
|
533 |
def create_loader(new_size, transformation, annotations1, annotations2=None,
|
534 |
+
batch_size=4, crop_prob=0.0, crop_fraction=0.7, min_objects=3,
|
535 |
+
h_flip_prob=0.0, v_flip_prob=0.0, max_rotate_deg=5, rotate_90_proba=0.0, rotate_proba=0.0,
|
536 |
+
seed=42, resize=True, keep_ratio=1, model_type='object'):
|
537 |
"""
|
538 |
Create a DataLoader for BPMN datasets with optional transformations and concatenation of two datasets.
|
539 |
|