Add augmentation script

training/pyproject.toml

@@ -7,6 +7,9 @@ dependencies = [
     "onnx>=1.15.0",
     "opencv-python>=4.9.0.80",
     "ultralytics>=8.1.10",
+    "albumentations>=1.4.4",
+    "datumaro>=1.6.1",
+    "loguru>=0.7.2",
 ]
 readme = "README.md"
 requires-python = ">= 3.8"

training/requirements-dev.lock

@@ -5,61 +5,266 @@
 #   pre: false
 #   features: []
 #   all-features: false
+#   with-sources: false
 
 -e file:.
+albumentations==1.4.7
+    # via training
+annotated-types==0.7.0
+    # via pydantic
+attrs==23.2.0
+    # via datumaro
+brotli==1.1.0
+    # via geventhttpclient
 certifi==2024.2.2
+    # via geventhttpclient
+    # via requests
+cffi==1.16.0
+    # via cryptography
 charset-normalizer==3.3.2
+    # via requests
 contourpy==1.2.0
+    # via matplotlib
+cryptography==42.0.7
+    # via datumaro
 cycler==0.12.1
+    # via matplotlib
+datumaro==1.6.1
+    # via training
+defusedxml==0.7.1
+    # via datumaro
 filelock==3.13.1
+    # via huggingface-hub
+    # via torch
 fonttools==4.48.1
+    # via matplotlib
 fsspec==2024.2.0
+    # via huggingface-hub
+    # via torch
+gevent==24.2.1
+    # via geventhttpclient
+geventhttpclient==2.3.1
+    # via tritonclient
+greenlet==3.0.3
+    # via gevent
+grpcio==1.64.0
+    # via ovmsclient
+    # via tritonclient
+h5py==3.11.0
+    # via datumaro
+huggingface-hub==0.23.1
+    # via tokenizers
 idna==3.6
+    # via requests
+imageio==2.34.1
+    # via scikit-image
+imagesize==1.4.1
+    # via datumaro
 jinja2==3.1.3
+    # via torch
+joblib==1.4.2
+    # via scikit-learn
+json-stream==2.3.2
+    # via datumaro
+json-stream-rs-tokenizer==0.4.26
+    # via json-stream
 kiwisolver==1.4.5
+    # via matplotlib
+lazy-loader==0.4
+    # via scikit-image
+loguru==0.7.2
+    # via training
+lxml==5.2.2
+    # via datumaro
 markupsafe==2.1.5
+    # via jinja2
 matplotlib==3.8.2
+    # via datumaro
+    # via pycocotools
+    # via seaborn
+    # via ultralytics
 mpmath==1.3.0
+    # via sympy
 mypy==1.8.0
 mypy-extensions==1.0.0
+    # via mypy
 networkx==3.2.1
+    # via datumaro
+    # via scikit-image
+    # via torch
+nibabel==5.2.1
+    # via datumaro
 numpy==1.26.4
-nvidia-cublas-cu12==12.1.3.1
-nvidia-cuda-cupti-cu12==12.1.105
-nvidia-cuda-nvrtc-cu12==12.1.105
-nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
-nvidia-cufft-cu12==11.0.2.54
-nvidia-curand-cu12==10.3.2.106
-nvidia-cusolver-cu12==11.4.5.107
-nvidia-cusparse-cu12==12.1.0.106
-nvidia-nccl-cu12==2.19.3
-nvidia-nvjitlink-cu12==12.3.101
-nvidia-nvtx-cu12==12.1.105
+    # via albumentations
+    # via contourpy
+    # via datumaro
+    # via h5py
+    # via imageio
+    # via matplotlib
+    # via nibabel
+    # via onnx
+    # via opencv-python
+    # via opencv-python-headless
+    # via openvino
+    # via ovmsclient
+    # via pandas
+    # via pyarrow
+    # via pycocotools
+    # via pyemd
+    # via scikit-image
+    # via scikit-learn
+    # via scipy
+    # via seaborn
+    # via shapely
+    # via tensorboardx
+    # via tifffile
+    # via torchvision
+    # via tritonclient
+    # via ultralytics
 onnx==1.15.0
+    # via training
 opencv-python==4.9.0.80
+    # via datumaro
+    # via training
+    # via ultralytics
+opencv-python-headless==4.9.0.80
+    # via albumentations
+openvino==2024.1.0
+    # via datumaro
+openvino-telemetry==2024.1.0
+    # via openvino
+orjson==3.10.0
+    # via datumaro
+ovmsclient==2023.1
+    # via datumaro
 packaging==23.2
+    # via huggingface-hub
+    # via lazy-loader
+    # via matplotlib
+    # via nibabel
+    # via openvino
+    # via scikit-image
+    # via tensorboardx
 pandas==2.2.0
+    # via datumaro
+    # via seaborn
+    # via ultralytics
 pillow==10.2.0
+    # via datumaro
+    # via imageio
+    # via matplotlib
+    # via scikit-image
+    # via torchvision
+    # via ultralytics
 protobuf==4.25.2
+    # via datumaro
+    # via onnx
+    # via ovmsclient
+    # via tensorboardx
+    # via tritonclient
 psutil==5.9.8
+    # via ultralytics
 py-cpuinfo==9.0.0
+    # via ultralytics
+pyarrow==16.1.0
+    # via datumaro
+pycocotools==2.0.7
+    # via datumaro
+pycparser==2.22
+    # via cffi
+pydantic==2.7.1
+    # via albumentations
+pydantic-core==2.18.2
+    # via pydantic
+pyemd==1.0.0
+    # via datumaro
 pyparsing==3.1.1
+    # via matplotlib
 python-dateutil==2.8.2
+    # via matplotlib
+    # via pandas
+python-rapidjson==1.17
+    # via tritonclient
 pytz==2024.1
+    # via pandas
 pyyaml==6.0.1
+    # via albumentations
+    # via datumaro
+    # via huggingface-hub
+    # via ultralytics
 requests==2.31.0
+    # via datumaro
+    # via huggingface-hub
+    # via ovmsclient
+    # via torchvision
+    # via ultralytics
+ruamel-yaml==0.18.6
+    # via datumaro
+ruamel-yaml-clib==0.2.8
+    # via ruamel-yaml
 ruff==0.2.1
+scikit-image==0.23.2
+    # via albumentations
+scikit-learn==1.5.0
+    # via albumentations
+    # via datumaro
 scipy==1.12.0
+    # via albumentations
+    # via datumaro
+    # via scikit-image
+    # via scikit-learn
+    # via ultralytics
 seaborn==0.13.2
+    # via ultralytics
+setuptools==70.0.0
+    # via zope-event
+    # via zope-interface
+shapely==2.0.4
+    # via datumaro
 six==1.16.0
+    # via python-dateutil
 sympy==1.12
+    # via torch
+tabulate==0.9.0
+    # via datumaro
+tensorboardx==2.6.2.2
+    # via datumaro
 thop==0.1.1.post2209072238
+    # via ultralytics
+threadpoolctl==3.5.0
+    # via scikit-learn
+tifffile==2024.5.10
+    # via scikit-image
+tokenizers==0.19.1
+    # via datumaro
 torch==2.2.0
+    # via thop
+    # via torchvision
+    # via ultralytics
 torchvision==0.17.0
+    # via ultralytics
 tqdm==4.66.1
-triton==2.2.0
+    # via datumaro
+    # via huggingface-hub
+    # via ultralytics
+tritonclient==2.17.0
+    # via datumaro
 typing-extensions==4.9.0
+    # via albumentations
+    # via datumaro
+    # via huggingface-hub
+    # via mypy
+    # via pydantic
+    # via pydantic-core
+    # via torch
 tzdata==2023.4
+    # via pandas
 ultralytics==8.1.10
+    # via training
 urllib3==2.2.0
+    # via geventhttpclient
+    # via requests
+zope-event==5.0
+    # via gevent
+zope-interface==6.4
+    # via gevent