BenjiELCA commited on
Commit
aa0b83f
·
1 Parent(s): 64b088f

correct create_loader

Browse files
Files changed (1) hide show
  1. modules/dataset_loader.py +3 -3
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.2, crop_fraction=0.7, min_objects=3,
535
- h_flip_prob=0.3, v_flip_prob=0.3, max_rotate_deg=20, rotate_90_proba=0.2, rotate_proba=0.3,
536
- seed=42, resize=True, keep_ratio=0.1, model_type='object'):
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