initial commit

.gitignore

@@ -0,0 +1,21 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+
+#system_bloat
+.DS_Store
+
+#miscellaneous
+dataset
+data
+docker-compose.yml
+ls-data
+mydata
+garbage-classification

.python-version

@@ -0,0 +1 @@
+3.12

main.py

@@ -0,0 +1,32 @@
+from pathlib import Path
+
+from ultralytics import YOLO
+
+
+DATASET_YAML = Path(__file__).parent / "dataset" / "dataset.yaml"
+RUNS_DIR = Path(__file__).parent / "runs"
+
+
+def train():
+    model = YOLO("yolo11n.pt")
+
+    results = model.train(
+        data=str(DATASET_YAML),
+        epochs=50,
+        imgsz=640,
+        batch=16,
+        project=str(RUNS_DIR),
+        name="pothole",
+        exist_ok=True,
+    )
+
+    print(f"Training complete. Results saved to: {results.save_dir}")
+    return results
+
+
+def main():
+    train()
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -0,0 +1,15 @@
+[project]
+name = "pothole-detection"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "kaggle>=2.0.0",
+    "ultralytics>=8.4.33",
+]
+
+[tool.uv.workspace]
+members = [
+    "dataset",
+]