Skip to content

Latest commit

 

History

History
78 lines (42 loc) · 4.16 KB

File metadata and controls

78 lines (42 loc) · 4.16 KB
graph LR
    Authentication_Manager["Authentication Manager"]
    Hub_Session_Manager["Hub Session Manager"]
    Hub_Utilities["Hub Utilities"]
    Model_Export_Functions["Model Export Functions"]
    doclayout_yolo_engine_model["doclayout_yolo.engine.model"]
    doclayout_yolo_data_utils["doclayout_yolo.data.utils"]
    Hub_Session_Manager -- "relies on" --> Authentication_Manager
    Hub_Session_Manager -- "utilizes" --> Hub_Utilities
    Model_Export_Functions -- "interacts with" --> Hub_Session_Manager
    doclayout_yolo_engine_model -- "interacts with" --> Hub_Session_Manager
    doclayout_yolo_data_utils -- "prepares data for" --> Hub_Session_Manager
Loading

CodeBoardingDemoContact

Details

The doclayout_yolo project's hub interaction subsystem is designed around a central Hub Session Manager that orchestrates all communications with external model hubs. This manager relies on the Authentication Manager for secure access and leverages Hub Utilities for efficient and reliable data transfer, especially for large files. Models prepared by Model Export Functions and potentially data processed by doclayout_yolo.data.utils are then handled by the Hub Session Manager for upload. The doclayout_yolo.engine.model component interacts with the Hub Session Manager to facilitate model training and inference operations involving the hub. This architecture ensures a clear separation of concerns, with data preparation and model handling distinct from the actual network session management.

Authentication Manager

Manages secure authentication with external model hubs, handling API keys, user credentials, and token management to ensure authorized access for all hub operations.

Related Classes/Methods:

Hub Session Manager

Orchestrates the primary interactions with the external hub, including loading pre-trained models, uploading trained models, and sending training metrics. It incorporates robust request handling (e.g., queuing, retries) for reliable communication.

Related Classes/Methods:

Hub Utilities

Provides general utility functions to support efficient and reliable communication with the hub, including smart request handling with progress tracking, crucial for large file transfers (models, datasets).

Related Classes/Methods:

Model Export Functions

Manages the preparation and export of models into various formats suitable for upload to the hub, including serialization and format conversion.

Related Classes/Methods:

doclayout_yolo.engine.model

Represents the model training/inference engine.

Related Classes/Methods:

doclayout_yolo.data.utils

Provides data utility functions, likely for data preparation and formatting.

Related Classes/Methods: