Trainer + Dataloaders (ultralytics#27)

Co-authored-by: Laughing-q <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <[email protected]> Co-authored-by: Ayush Chaurasia <[email protected]> Co-authored-by: Ayush Chaurasia <[email protected]>
swbycv · Oct 10, 2022 · d0b3c98 · d0b3c98
1 parent 7a2e5fd
commit d0b3c98
Show file tree

Hide file tree

Showing 27 changed files with 2,885 additions and 9 deletions.
diff --git a/.gitignore b/.gitignore
@@ -127,3 +127,7 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+# datasets and projects
+datasets/
+ultralytics-yolo/
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1,12 +1,5 @@
-# Include the README
 include *.md
 include requirements.txt
-
-# Include the license file
 include LICENSE
-
-# Include setup.py
 include setup.py
-
-# Include the data files
-recursive-include data *
+recursive-include ultralytics *.yaml
diff --git a/requirements.txt b/requirements.txt
@@ -3,6 +3,7 @@
 
 # Base ----------------------------------------
 fire>=0.4.0
+hydra-core>=1.2.0
 matplotlib>=3.2.2
 numpy>=1.18.5
 opencv-python>=4.1.1
@@ -44,4 +45,3 @@ thop>=0.1.1  # FLOPs computation
 
 # HUB -----------------------------------------
 GitPython>=3.1.24
-requests
diff --git a/ultralytics/yolo/__init__.py b/ultralytics/yolo/__init__.py
@@ -0,0 +1,3 @@
+from .engine.trainer import BaseTrainer
+
+__all__ = ["BaseTrainer"]  # allow simpler import
diff --git a/ultralytics/yolo/data/__init__.py b/ultralytics/yolo/data/__init__.py
@@ -0,0 +1,3 @@
+from .build import build_classification_dataloader, build_dataloader
+from .dataset import ClassificationDataset, SemanticDataset, YOLODataset
+from .dataset_wrappers import MixAndRectDataset
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		from .engine.trainer import BaseTrainer

		__all__ = ["BaseTrainer"] # allow simpler import