Skip to content

Latest commit

 

History

History
68 lines (37 loc) · 3.29 KB

File metadata and controls

68 lines (37 loc) · 3.29 KB
graph LR
    ImageBaseDatasetHandler["ImageBaseDatasetHandler"]
    SpecificDatasetImplementations["SpecificDatasetImplementations"]
    DataAugmentationModule["DataAugmentationModule"]
    ROMPGeneralUtilities["ROMPGeneralUtilities"]
    Trace2VideoDataLoader["Trace2VideoDataLoader"]
    SpecificDatasetImplementations -- "implements/extends" --> ImageBaseDatasetHandler
    ImageBaseDatasetHandler -- "utilizes" --> DataAugmentationModule
    ImageBaseDatasetHandler -- "leverages" --> ROMPGeneralUtilities
    Trace2VideoDataLoader -- "leverages" --> ROMPGeneralUtilities
    Trace2VideoDataLoader -- "prepares data for" --> DataAugmentationModule
Loading

CodeBoardingDemoContact

Details

Data Input & Preprocessing

ImageBaseDatasetHandler

Serves as the abstract base for handling single-frame image datasets, defining the interface and common logic for loading, standardizing, and preparing individual image frames for subsequent processing.

Related Classes/Methods:

SpecificDatasetImplementations

Concrete dataset classes that inherit from or utilize ImageBaseDatasetHandler, providing dataset-specific logic for parsing annotations and locating files for various image datasets.

Related Classes/Methods:

DataAugmentationModule

Provides a comprehensive set of functions for augmenting image and pose data, crucial for increasing training data diversity and model robustness by applying various transformations.

Related Classes/Methods:

ROMPGeneralUtilities

Offers general utility functions for the ROMP project, including fundamental image preprocessing, temporal smoothing, and extracting frame paths, which are leveraged across different data handling components.

Related Classes/Methods:

Trace2VideoDataLoader

Specializes in loading and preparing video sequences specifically for the trace2 model, handling the complexities of video input and ensuring readiness for sequential model processing.

Related Classes/Methods: