81bbf66aab
* Add project code * Logger improvements * Improvements to web demo code * added create_wlasl_landmarks_dataset.py and xtract_mediapipe_landmarks.py * Fix rotation augmentation * fixed error in docstring, and removed unnecessary replace -1 -> 0 * Readme updates * Share base notebooks * Add notebooks and unify for different datasets * requirements update * fixes * Make evaluate more deterministic * Allow training with clearml * refactor preprocessing and apply linter * Minor fixes * Minor notebook tweaks * Readme updates * Fix PR comments * Remove unneeded code * Add banner to Readme --------- Co-authored-by: Gabriel Lema <gabriel.lema@xmartlabs.com>
18 lines
482 B
Python
18 lines
482 B
Python
|
|
import os
|
|
|
|
|
|
class DatasetLoader():
|
|
"""Abstract class that serves to load datasets from different sources (local, ClearML, other tracker)
|
|
"""
|
|
|
|
def get_dataset_folder(self, dataset_project, dataset_name):
|
|
return NotImplementedError()
|
|
|
|
|
|
class LocalDatasetLoader(DatasetLoader):
|
|
|
|
def get_dataset_folder(self, dataset_project, dataset_name):
|
|
base_folder = os.environ.get("BASE_DATA_FOLDER", "data")
|
|
return os.path.join(base_folder, dataset_name)
|