Skip to content

Instantly share code, notes, and snippets.

@AlessandroMondin

AlessandroMondin/training_dataset_init.py

Last active December 23, 2022 14:34
Show Gist options
  • Save AlessandroMondin/4a31f32b6f1f8cceab10016138cf8146 to your computer and use it in GitHub Desktop.
Save AlessandroMondin/4a31f32b6f1f8cceab10016138cf8146 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
training_dataset_init.py
class Training_Dataset(Dataset):
def __init__(self,
root_directory=config.ROOT_DIR,
transform=None,
train=True,
rect_training=False,
default_size=640,
bs=64,
bboxes_format="coco",
ultralytics_loss=False,
):
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment