Package `data_handling`

Expand source code

# from .conversion import *
# from .ffmpeg_utils import *
# from .modifications import *
# from .split import *
# from .utils import *

# from .coco_assistant.coco_assistant import *

Sub-modules

data_handling.conversion: Functions to convert between coco to yolo format and vice versa
data_handling.ffmpeg_utils: Supports splitting video into frames and combining a folder of images.
data_handling.modifications: Functions to do modifications on the dataset like removing small annotations, applying augmentations, RATT, or using SimCLR to generate diverse data …
data_handling.split: Functions to split data into training and validation sets on both coco and yolo formats as well as creating a new annotation file for a given subset …
data_handling.utils: Utility functions for data handling like: 1. Drawing annotations 2. Getting stats. 3. Removing categories or creating subset of dataset of particular categories. 4. Merging datasets
data_handling.benchmark: Function to calculate confusion matrix values from data