diff --git a/.DS_Store b/.DS_Store deleted file mode 100644 index 36c3335..0000000 Binary files a/.DS_Store and /dev/null differ diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..9429ef9 --- /dev/null +++ b/.gitattributes @@ -0,0 +1 @@ +training/weights_* filter=lfs diff=lfs merge=lfs -text diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..a96be9c --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +*/.DS_Store +*/__pycache__/ diff --git a/UnEye.ipynb b/bin/UnEye.ipynb similarity index 100% rename from UnEye.ipynb rename to bin/UnEye.ipynb diff --git a/UnEye.py b/bin/UnEye.py similarity index 100% rename from UnEye.py rename to bin/UnEye.py diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..ab920e6 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,19 @@ +[tool.poetry] +name = "uneye" +version = "0.1.0" +description = "" +authors = [] +readme = "README.md" + +[tool.poetry.dependencies] +python = "^3.11" +torch = "^2.6.0" +numpy = "^1.26.0" +scipy = "~1.14.0" +scikit-image = "^0.25.0" +scikit-learn = "^1.6.0" + + +[build-system] +requires = ["poetry-core"] +build-backend = "poetry.core.masonry.api" diff --git a/requirements_lin.txt b/requirements_lin.txt deleted file mode 100644 index 2aeda17..0000000 --- a/requirements_lin.txt +++ /dev/null @@ -1,8 +0,0 @@ -http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl -torchvision -numpy -scipy -scikit-image -sklearn -jupyter -matplotlib diff --git a/requirements_mac.txt b/requirements_mac.txt deleted file mode 100644 index b388104..0000000 --- a/requirements_mac.txt +++ /dev/null @@ -1,8 +0,0 @@ -torch -torchvision -numpy -scipy -scikit-image -sklearn -jupyter -matplotlib diff --git a/requirements_wind.txt b/requirements_wind.txt deleted file mode 100644 index e190278..0000000 --- a/requirements_wind.txt +++ /dev/null @@ -1,8 +0,0 @@ -http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl -torchvision -numpy -scipy -scikit-image -sklearn -jupyter -matplotlib diff --git a/setup.py b/setup.py deleted file mode 100644 index 00da9ac..0000000 --- a/setup.py +++ /dev/null @@ -1,7 +0,0 @@ -#!/usr/bin/python3 - -from distutils.core import setup -setup(name='uneye', - version='0.1', - packages=['uneye'], - ) diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/uneye/__init__.py b/uneye/__init__.py index f97b7d4..f119832 100644 --- a/uneye/__init__.py +++ b/uneye/__init__.py @@ -1 +1,7 @@ from .classifier import DNN + + +def weights_dir(): + from pathlib import Path + + return Path(__file__).parent / "weights" diff --git a/training/notes.md b/uneye/weights/notes.md similarity index 100% rename from training/notes.md rename to uneye/weights/notes.md diff --git a/training/weights_1+2+3 b/uneye/weights/weights_1+2+3 similarity index 100% rename from training/weights_1+2+3 rename to uneye/weights/weights_1+2+3 diff --git a/training/weights_Andersson b/uneye/weights/weights_Andersson similarity index 100% rename from training/weights_Andersson rename to uneye/weights/weights_Andersson diff --git a/training/weights_dataset1 b/uneye/weights/weights_dataset1 similarity index 100% rename from training/weights_dataset1 rename to uneye/weights/weights_dataset1 diff --git a/training/weights_dataset2 b/uneye/weights/weights_dataset2 similarity index 100% rename from training/weights_dataset2 rename to uneye/weights/weights_dataset2 diff --git a/training/weights_dataset3 b/uneye/weights/weights_dataset3 similarity index 100% rename from training/weights_dataset3 rename to uneye/weights/weights_dataset3 diff --git a/training/weights_synthetic b/uneye/weights/weights_synthetic similarity index 100% rename from training/weights_synthetic rename to uneye/weights/weights_synthetic