qubvel · omidvarnia · Feb 10, 2025
diff --git a/examples/binary segmentation (camvid).ipynb b/examples/binary segmentation (camvid).ipynb
diff --git a/examples/binary_segmentation_intro.py b/examples/binary_segmentation_intro.py
diff --git a/output_masks/0.png b/output_masks/0.png
diff --git a/output_masks/0_2.png b/output_masks/0_2.png
diff --git a/output_masks/1.png b/output_masks/1.png
diff --git a/output_masks/1_2.png b/output_masks/1_2.png
diff --git a/output_masks/2.png b/output_masks/2.png
diff --git a/output_masks/2_2.png b/output_masks/2_2.png
diff --git a/output_masks/3.png b/output_masks/3.png
diff --git a/output_masks/3_2.png b/output_masks/3_2.png
diff --git a/output_masks/4.png b/output_masks/4.png
diff --git a/output_masks/4_2.png b/output_masks/4_2.png
diff --git a/output_masks/5.png b/output_masks/5.png
diff --git a/output_masks/5_2.png b/output_masks/5_2.png
diff --git a/output_masks/6.png b/output_masks/6.png
diff --git a/output_masks/6_2.png b/output_masks/6_2.png
diff --git a/output_masks/7.png b/output_masks/7.png
diff --git a/output_masks/7_2.png b/output_masks/7_2.png
diff --git a/output_masks/8_2.png b/output_masks/8_2.png
diff --git a/unet_segmentation_requirements.txt b/unet_segmentation_requirements.txt
@@ -0,0 +1,149 @@
+absl-py==2.1.0
+aiohappyeyeballs==2.4.6
+aiohttp==3.11.12
+aiosignal==1.3.2
+albucore==0.0.23
+albumentations==2.0.3
+annotated-types==0.7.0
+appdirs==1.4.4
+array-record==0.6.0
+asttokens==3.0.0
+astunparse==1.6.3
+attrs==25.1.0
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+comm==0.2.2
+contourpy==1.3.1
+cycler==0.12.1
+debugpy==1.8.12
+decorator==5.1.1
+dm-tree==0.1.9
+docstring-parser==0.16
+efficientnet==1.0.0
+etils==1.11.0
+executing==2.2.0
+filelock==3.17.0
+flatbuffers==25.1.24
+fonttools==4.56.0
+frozenlist==1.5.0
+fsspec==2025.2.0
+gast==0.6.0
+google-pasta==0.2.0
+googleapis-common-protos==1.66.0
+grpcio==1.70.0
+h5py==3.12.1
+huggingface-hub==0.28.1
+idna==3.10
+image-classifiers==1.0.0
+imageio==2.37.0
+immutabledict==4.2.1
+importlib-resources==6.5.2
+ipykernel==6.29.5
+ipython==8.32.0
+jedi==0.19.2
+jinja2==3.1.5
+jupyter-client==8.6.3
+jupyter-core==5.7.2
+keras==3.8.0
+keras-applications==1.0.8
+kiwisolver==1.4.8
+lazy-loader==0.4
+libclang==18.1.1
+lightning==2.5.0.post0
+lightning-utilities==0.12.0
+markdown==3.7
+markdown-it-py==3.0.0
+markupsafe==3.0.2
+matplotlib==3.10.0
+matplotlib-inline==0.1.7
+mdurl==0.1.2
+ml-dtypes==0.4.1
+mpmath==1.3.0
+multidict==6.1.0
+namex==0.0.8
+nest-asyncio==1.6.0
+networkx==3.4.2
+numpy==2.0.2
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvcc-cu12==12.5.82
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-cusparselt-cu12==0.6.2
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
+opencv-python==4.11.0.86
+opencv-python-headless==4.11.0.86
+opt-einsum==3.4.0
+optree==0.14.0
+packaging==24.2
+parso==0.8.4
+pexpect==4.9.0
+pillow==11.1.0
+pip==24.0
+platformdirs==4.3.6
+promise==2.3
+prompt-toolkit==3.0.50
+propcache==0.2.1
+protobuf==5.29.3
+psutil==6.1.1
+ptpython==3.0.29
+ptyprocess==0.7.0
+pure-eval==0.2.3
+pyarrow==19.0.0
+pydantic==2.10.6
+pydantic-core==2.27.2
+pygments==2.19.1
+pyparsing==3.2.1
+python-dateutil==2.9.0.post0
+pytorch-lightning==2.5.0.post0
+pyyaml==6.0.2
+pyzmq==26.2.1
+requests==2.32.3
+rich==13.9.4
+safetensors==0.5.2
+scikit-image==0.25.1
+scipy==1.15.1
+segmentation-models==1.0.1
+segmentation-models-pytorch @ git+https://github.com/qubvel/segmentation_models.pytorch@5c5413666ade180970af211d92b9d2e207e7b6b9
+setuptools==65.5.0
+simple-parsing==0.1.7
+simsimd==6.2.1
+six==1.17.0
+stack-data==0.6.3
+stringzilla==3.11.3
+sympy==1.13.1
+tensorboard==2.18.0
+tensorboard-data-server==0.7.2
+tensorflow==2.18.0
+tensorflow-advanced-segmentation-models @ git+https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models.git@3714839ee49759b26e2b0ae3d3a0aa37b00df962
+tensorflow-datasets==4.9.7
+tensorflow-io-gcs-filesystem==0.37.1
+tensorflow-metadata==1.16.1
+termcolor==2.5.0
+tfds-nightly==4.9.7.dev202502080044
+tifffile==2025.1.10
+timm==1.0.14
+toml==0.10.2
+torch==2.6.0
+torchmetrics==1.6.1
+torchvision==0.21.0
+tornado==6.4.2
+tqdm==4.67.1
+traitlets==5.14.3
+triton==3.2.0
+typing-extensions==4.12.2
+urllib3==2.3.0
+wcwidth==0.2.13
+werkzeug==3.1.3
+wheel==0.45.1
+wrapt==1.17.2
+yarl==1.18.3
+zipp==3.21.0