diff --git a/.gitattributes b/.gitattributes index a6344aac8c09253b3b630fb776ae94478aa0275b..b403f3045efe0c73901074d4e3f3bdedc85e7937 100644 --- a/.gitattributes +++ b/.gitattributes @@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text *.zip filter=lfs diff=lfs merge=lfs -text *.zst filter=lfs diff=lfs merge=lfs -text *tfevents* filter=lfs diff=lfs merge=lfs -text +UNet++.[[:space:]]A[[:space:]]Nested[[:space:]]U-Net[[:space:]]Architecture[[:space:]]for[[:space:]]Medical[[:space:]]Image[[:space:]]Segmentation[[:space:]](slides).pdf filter=lfs diff=lfs merge=lfs -text +UNet++.[[:space:]]Redesigning[[:space:]]Skip[[:space:]]Connections[[:space:]]to[[:space:]]Exploit[[:space:]]Multiscale[[:space:]]Features[[:space:]]in[[:space:]]Image[[:space:]]Segmentation[[:space:]]+[[:space:]]Supplement.pdf filter=lfs diff=lfs merge=lfs -text +UNet++.[[:space:]]Redesigning[[:space:]]Skip[[:space:]]Connections[[:space:]]to[[:space:]]Exploit[[:space:]]Multiscale[[:space:]]Features[[:space:]]in[[:space:]]Image[[:space:]]Segmentation.pdf filter=lfs diff=lfs merge=lfs -text diff --git a/UNet++. A Nested U-Net Architecture for Medical Image Segmentation (slides).pdf b/UNet++. A Nested U-Net Architecture for Medical Image Segmentation (slides).pdf new file mode 100644 index 0000000000000000000000000000000000000000..4a644a4a3553970e7060207680858bb92ed928c8 --- /dev/null +++ b/UNet++. A Nested U-Net Architecture for Medical Image Segmentation (slides).pdf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:16c8d6f36269b9fde18f28e44640e4d9cbff4d181bd4665f516f179208a9cb97 +size 25878007 diff --git a/UNet++. Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentation + Supplement.pdf b/UNet++. Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentation + Supplement.pdf new file mode 100644 index 0000000000000000000000000000000000000000..d9947dc60599f706590dba4c90cf3e9d8d458b0d --- /dev/null +++ b/UNet++. Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentation + Supplement.pdf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:17a507f6309fbc7e490eb10396f2f897c1d2ab17f1f2f55af9e2940d7e296d4e +size 1783364 diff --git a/UNet++. Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentation.pdf b/UNet++. Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentation.pdf new file mode 100644 index 0000000000000000000000000000000000000000..ab25e402770f752a632ea42a1e3358795b24c2a5 --- /dev/null +++ b/UNet++. Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentation.pdf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d57f5a33d0444a9b60c142c5bbf1e66726b4ab819131d012132dd4ab98064cbe +size 3518768 diff --git a/code/UNetPlusPlus [cjjsu] +1.zip b/code/UNetPlusPlus [cjjsu] +1.zip new file mode 100644 index 0000000000000000000000000000000000000000..35b268eb4819af071acaa0b25b3d816c18d39ab2 --- /dev/null +++ b/code/UNetPlusPlus [cjjsu] +1.zip @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c4f411e9f7b5664b3c194ef0ca5d454ad6aee7cff0500020ab5b3fca5fb044d +size 18376869 diff --git a/code/UNetPlusPlus.zip b/code/UNetPlusPlus.zip new file mode 100644 index 0000000000000000000000000000000000000000..cf7c41a5a7ce2d4ba19789d4217f322ed36a116c --- /dev/null +++ b/code/UNetPlusPlus.zip @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9dd4c34e6bdea2fe27cd1fd9e48721bcbcb386dce19aee9c38832594785c96d1 +size 13490664 diff --git a/models/UNET_plus_plus_Brain_segmentation/.gitattributes b/models/UNET_plus_plus_Brain_segmentation/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/UNET_plus_plus_Brain_segmentation/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/UNET_plus_plus_Brain_segmentation/README.md b/models/UNET_plus_plus_Brain_segmentation/README.md new file mode 100644 index 0000000000000000000000000000000000000000..c5e69ab35b32d6454d769f62b6b91553fe6187ee --- /dev/null +++ b/models/UNET_plus_plus_Brain_segmentation/README.md @@ -0,0 +1,16 @@ +--- +license: mit +datasets: +- rishirajbal/UNET_plus_plus_BRAIN_DATA +language: +- en +metrics: +- accuracy +base_model: +- rishirajbal/UNET_plus_plus_Brain_segmentation +pipeline_tag: image-segmentation +library_name: keras +tags: +- medical +new_version: rishirajbal/UNET_plus_plus_Brain_segmentation +--- \ No newline at end of file diff --git a/models/UNET_plus_plus_Brain_segmentation/source.txt b/models/UNET_plus_plus_Brain_segmentation/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..7e3aa106aeccdb42cbf5777327b681668e97c2c8 --- /dev/null +++ b/models/UNET_plus_plus_Brain_segmentation/source.txt @@ -0,0 +1 @@ +https://huggingface.co/rishirajbal/UNET_plus_plus_Brain_segmentation \ No newline at end of file diff --git a/models/UNET_plus_plus_Brain_segmentation/unet_model.h5 b/models/UNET_plus_plus_Brain_segmentation/unet_model.h5 new file mode 100644 index 0000000000000000000000000000000000000000..3a6504ed70d19e1eb0d3a87ed4e52678fe8d1fd9 --- /dev/null +++ b/models/UNET_plus_plus_Brain_segmentation/unet_model.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8674ba558edc44f9be08ca79d94eb633c488c2c8bb06014e02bbe5f628d4e929 +size 637233872 diff --git a/models/UnetPlusPlus_resnet34/.gitattributes b/models/UnetPlusPlus_resnet34/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/UnetPlusPlus_resnet34/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/UnetPlusPlus_resnet34/README.md b/models/UnetPlusPlus_resnet34/README.md new file mode 100644 index 0000000000000000000000000000000000000000..7be5fc7f47d5db027d120b8024982df93db95b74 --- /dev/null +++ b/models/UnetPlusPlus_resnet34/README.md @@ -0,0 +1,3 @@ +--- +license: mit +--- diff --git a/models/UnetPlusPlus_resnet34/model.pth b/models/UnetPlusPlus_resnet34/model.pth new file mode 100644 index 0000000000000000000000000000000000000000..c234509129da487266d82015644ce1ae2a7a58cf --- /dev/null +++ b/models/UnetPlusPlus_resnet34/model.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:23d302008ef8bd47ed70a6996345cac6c2efd30459082797be16a88333321cda +size 313261125 diff --git a/models/UnetPlusPlus_resnet34/source.txt b/models/UnetPlusPlus_resnet34/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..fd71366472fa1684177e5a2d2fee2f4683fe75b6 --- /dev/null +++ b/models/UnetPlusPlus_resnet34/source.txt @@ -0,0 +1 @@ +https://huggingface.co/anhtu77/UnetPlusPlus_resnet34/tree/main \ No newline at end of file diff --git a/models/Unetplusplus_dolphin_segmentation/.gitattributes b/models/Unetplusplus_dolphin_segmentation/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/Unetplusplus_dolphin_segmentation/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/Unetplusplus_dolphin_segmentation/README.md b/models/Unetplusplus_dolphin_segmentation/README.md new file mode 100644 index 0000000000000000000000000000000000000000..ab1026189d4daa3c13229a3ffce8d7b35a755eb3 --- /dev/null +++ b/models/Unetplusplus_dolphin_segmentation/README.md @@ -0,0 +1,3 @@ +--- +license: unknown +--- diff --git a/models/Unetplusplus_dolphin_segmentation/Unetplusplusdolphinseg.pt b/models/Unetplusplus_dolphin_segmentation/Unetplusplusdolphinseg.pt new file mode 100644 index 0000000000000000000000000000000000000000..d5880e4ceec35905ac17ba0e896d00b18d25bef3 --- /dev/null +++ b/models/Unetplusplus_dolphin_segmentation/Unetplusplusdolphinseg.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21ee6b78f1f53d91e1b0b47e0930b94914dae325d2b416896839aec3d292f2ed +size 84072388 diff --git a/models/Unetplusplus_dolphin_segmentation/source.txt b/models/Unetplusplus_dolphin_segmentation/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..08441e6fe4a698b85f969548a89346b45f66b8c2 --- /dev/null +++ b/models/Unetplusplus_dolphin_segmentation/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Alexandr45/Unetplusplus_dolphin_segmentation \ No newline at end of file diff --git a/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/.gitattributes b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/README.md b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/README.md new file mode 100644 index 0000000000000000000000000000000000000000..23bb8239b30ad636a1b592db0346c4753dc5325d --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/README.md @@ -0,0 +1,9 @@ +--- +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +--- + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration: +- Library: [More Information Needed] +- Docs: [More Information Needed] \ No newline at end of file diff --git a/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/config.json b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/config.json new file mode 100644 index 0000000000000000000000000000000000000000..90f0cdb93f957958c94d2e1c3a583df003d3bd33 --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/config.json @@ -0,0 +1,6 @@ +{ + "arch": "unetplusplus", + "classes": 3, + "encoder": "mobilenetv3_small_050", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/model.safetensors b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..1dd56a6e6738442ac65bd1f0390fcb73b4f870f4 --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f1b302fb605cd91356361977aeabd1b7d0091f2d80a99c1d5fbd9a16c782833c +size 9188060 diff --git a/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/source.txt b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..074ba2005e3c13bba962fc082cf93e3f8dfbde93 --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050/source.txt @@ -0,0 +1 @@ +https://huggingface.co/ClementP/fundus-odmac-segmentation-unetplusplus-mobilenetv3_small_050 \ No newline at end of file diff --git a/models/fundus-odmac-segmentation-unetplusplus-seresnet50/.gitattributes b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/fundus-odmac-segmentation-unetplusplus-seresnet50/README.md b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/README.md new file mode 100644 index 0000000000000000000000000000000000000000..23bb8239b30ad636a1b592db0346c4753dc5325d --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/README.md @@ -0,0 +1,9 @@ +--- +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +--- + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration: +- Library: [More Information Needed] +- Docs: [More Information Needed] \ No newline at end of file diff --git a/models/fundus-odmac-segmentation-unetplusplus-seresnet50/config.json b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fece970d9bb620b09db33242fb1f24fcb8081894 --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/config.json @@ -0,0 +1,6 @@ +{ + "arch": "unetplusplus", + "classes": 3, + "encoder": "seresnet50", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/fundus-odmac-segmentation-unetplusplus-seresnet50/model.safetensors b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..6faf97e3a6a5f8b67b5f92ad08b22a94c5933441 --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7f214244ddbf50fa1c149a3d2a711886208f6e1f5f31d776bda173b47a942790 +size 206366596 diff --git a/models/fundus-odmac-segmentation-unetplusplus-seresnet50/source.txt b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..de14eec8fa6716b33827c77f571e6d2fe512d9c1 --- /dev/null +++ b/models/fundus-odmac-segmentation-unetplusplus-seresnet50/source.txt @@ -0,0 +1 @@ +https://huggingface.co/ClementP/fundus-odmac-segmentation-unetplusplus-seresnet50 \ No newline at end of file diff --git a/models/octa-unetplusplus/.gitattributes b/models/octa-unetplusplus/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..dab9a4e17afd2ef39d90ccb0b40ef2786fe77422 --- /dev/null +++ b/models/octa-unetplusplus/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/octa-unetplusplus/model_fold0/config.json b/models/octa-unetplusplus/model_fold0/config.json new file mode 100644 index 0000000000000000000000000000000000000000..0a7f25582e1e0a54a5737a9610cfa5e987611f22 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m4_0" + ], + "val_imgs": [ + "m4_0" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold0", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-chiivm0-609" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold0/model_best_f0.pt b/models/octa-unetplusplus/model_fold0/model_best_f0.pt new file mode 100644 index 0000000000000000000000000000000000000000..9cd82b326d087f70e228afc19838eeb0e3e7534b --- /dev/null +++ b/models/octa-unetplusplus/model_fold0/model_best_f0.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7f6079a1cbb9025b10254343e11d8a9573a969e5df33d551e7c4e587cf5b4184 +size 379775475 diff --git a/models/octa-unetplusplus/model_fold0/model_last.pt b/models/octa-unetplusplus/model_fold0/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..053d22ce51f0b981ba10dc1ad4adbba4c8f28f94 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c53ea81dce6a03d27035a2ea162ce6232fe770e7ee03dc173b992f13943dc8d +size 379774439 diff --git a/models/octa-unetplusplus/model_fold0_crop96/config.json b/models/octa-unetplusplus/model_fold0_crop96/config.json new file mode 100644 index 0000000000000000000000000000000000000000..97200b4cb839ebcab15465b71f17f6b9fae8ef16 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_crop96/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations", + "test_imgs": [ + "m4_0" + ], + "val_imgs": [ + "m4_0" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold0_test_moreparams_crop96", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.1, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 96, + 96, + 96 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.5, + 1.2 + ], + "intensity_shift_offsets": [ + 0.05, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "7a8f7f7", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computegpu-5" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold0_crop96/model_best_f0.pt b/models/octa-unetplusplus/model_fold0_crop96/model_best_f0.pt new file mode 100644 index 0000000000000000000000000000000000000000..421bfd898f8bf3a200c33a732eeb3cd2b7338905 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_crop96/model_best_f0.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f74df7404cd83c46ccd658155905ec3dd82d66e6371bbc78b9e95ea0ad6852f2 +size 379775411 diff --git a/models/octa-unetplusplus/model_fold0_crop96/model_last.pt b/models/octa-unetplusplus/model_fold0_crop96/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..c48853f38503568f9e32a25ad804be69490a134f --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_crop96/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:02eb51439fb9b51f2ae5c5799351728225dae11b16fbfb4837f9faa255412788 +size 379774439 diff --git a/models/octa-unetplusplus/model_fold0_doppler/config.json b/models/octa-unetplusplus/model_fold0_doppler/config.json new file mode 100644 index 0000000000000000000000000000000000000000..87a492e5a1ed6964e84d4b92fa0e56a548471f97 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_doppler/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m4_0" + ], + "val_imgs": [ + "m4_0" + ], + "train_on_real": true, + "use_doppler": true, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold0_test_moreparams_doppler", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computeibmgpu-vesta11" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold0_doppler/model_best_f0.pt b/models/octa-unetplusplus/model_fold0_doppler/model_best_f0.pt new file mode 100644 index 0000000000000000000000000000000000000000..7c9d705f46313cdbee43c87eee63c161b3420b68 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_doppler/model_best_f0.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:04bdfd5625b5d77d04df28c011d2416ad89da350c100ac89aa6baffc2842e87f +size 379837619 diff --git a/models/octa-unetplusplus/model_fold0_doppler/model_last.pt b/models/octa-unetplusplus/model_fold0_doppler/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..3aea864f924d946424d7971189d37297bb212c9c --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_doppler/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6472cb53ffe62e2800bc81326b0a11a3f4a3b1b3fb4f03ae7593d6d1707250b +size 379836647 diff --git a/models/octa-unetplusplus/model_fold0_intaug/config.json b/models/octa-unetplusplus/model_fold0_intaug/config.json new file mode 100644 index 0000000000000000000000000000000000000000..1fa40db0214f80544159239ead45660ff0cf96fe --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_intaug/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m4_0" + ], + "val_imgs": [ + "m4_0" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold0_test_moreparams_intaug", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0.5, + "p_intensity_shift": 0.5, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computeibmgpu-vesta9" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold0_intaug/model_best_f0.pt b/models/octa-unetplusplus/model_fold0_intaug/model_best_f0.pt new file mode 100644 index 0000000000000000000000000000000000000000..13497e9f7e8dd056a67dd02e4312b7416bd01b30 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_intaug/model_best_f0.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d4dfd557861fc8ac5c553456890675c22955385d037bac0848e336d078cf19e5 +size 379775411 diff --git a/models/octa-unetplusplus/model_fold0_intaug/model_last.pt b/models/octa-unetplusplus/model_fold0_intaug/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..e24621e2c96e943730831ca28bc099b5641d8c93 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_intaug/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a903d671cc1dee280c169fdcf86e4952d7be386b00211fd7f7a888c66ae64da7 +size 379774439 diff --git a/models/octa-unetplusplus/model_fold0_pre/config.json b/models/octa-unetplusplus/model_fold0_pre/config.json new file mode 100644 index 0000000000000000000000000000000000000000..81992a2ebed3a6fef9ddec1380649127b3b253dc --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_pre/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m4_0" + ], + "val_imgs": [ + "m4_0" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold0_pre", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computeibmgpu-vesta10" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold0_pre/model_best_f0.pt b/models/octa-unetplusplus/model_fold0_pre/model_best_f0.pt new file mode 100644 index 0000000000000000000000000000000000000000..28a7e0b6b5738137de08cf26349c6b28584c2106 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_pre/model_best_f0.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c39a138ed1e0d5eea9259998aa8434913b274e10a635e39698c05b206c6c0b59 +size 379775411 diff --git a/models/octa-unetplusplus/model_fold0_pre/model_last.pt b/models/octa-unetplusplus/model_fold0_pre/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..69b12c8d65da310cf737a311c874a847676dc002 --- /dev/null +++ b/models/octa-unetplusplus/model_fold0_pre/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:895fd93ae7e10cca793e8d9cdda59f8ee9e841edc17335fb0f2008c8429f0e6d +size 379774439 diff --git a/models/octa-unetplusplus/model_fold1/config.json b/models/octa-unetplusplus/model_fold1/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fafb7406f036ad3ff44312cb6f4c1209a92e5a36 --- /dev/null +++ b/models/octa-unetplusplus/model_fold1/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m4_1" + ], + "val_imgs": [ + "m4_1" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_0", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold1", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-chiivm0-605" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold1/model_best_f1.pt b/models/octa-unetplusplus/model_fold1/model_best_f1.pt new file mode 100644 index 0000000000000000000000000000000000000000..6ace3b66954ab4d78d9a56a4592453e94000fcae --- /dev/null +++ b/models/octa-unetplusplus/model_fold1/model_best_f1.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fa0a72a12d6fcfafbad8de168e0c76d719c2cae341e16431fd9bc0ff26c55235 +size 379775475 diff --git a/models/octa-unetplusplus/model_fold1/model_last.pt b/models/octa-unetplusplus/model_fold1/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..00b63eb6b2124aedf886fc53472d1dff80f2a37c --- /dev/null +++ b/models/octa-unetplusplus/model_fold1/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:afd0cf8236ded7d4d0bb40a7f3274e7af8965250619ff1d531c7941102f67e90 +size 379774439 diff --git a/models/octa-unetplusplus/model_fold1_pre/config.json b/models/octa-unetplusplus/model_fold1_pre/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5dab7eec78643f870eeb0f02fc1dc628a65bfbb6 --- /dev/null +++ b/models/octa-unetplusplus/model_fold1_pre/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m4_1" + ], + "val_imgs": [ + "m4_1" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_0", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold1_pre", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computegpu-5" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold1_pre/model_best_f1.pt b/models/octa-unetplusplus/model_fold1_pre/model_best_f1.pt new file mode 100644 index 0000000000000000000000000000000000000000..f266af32b3320c50ab52429999f6dfaf67220e68 --- /dev/null +++ b/models/octa-unetplusplus/model_fold1_pre/model_best_f1.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f19073f01d88990c0b7d03ee6aeb6217b4bda8c2f69617a1592169cd86fb9192 +size 379775411 diff --git a/models/octa-unetplusplus/model_fold1_pre/model_last.pt b/models/octa-unetplusplus/model_fold1_pre/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..18a1368fcecf025955b4b36767dbce500d60ab67 --- /dev/null +++ b/models/octa-unetplusplus/model_fold1_pre/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01253a4ab86e307df89a1263a370848c9829d834c815dfdbe27b67a9c2e81844 +size 379774439 diff --git a/models/octa-unetplusplus/model_fold2/config.json b/models/octa-unetplusplus/model_fold2/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f1d546ca4672931ac542f7591f4ffb9e2772025b --- /dev/null +++ b/models/octa-unetplusplus/model_fold2/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m44_0" + ], + "val_imgs": [ + "m44_0" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m4_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold2", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-chiivm0-606" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold2/model_best_f2.pt b/models/octa-unetplusplus/model_fold2/model_best_f2.pt new file mode 100644 index 0000000000000000000000000000000000000000..4c601fc6873302338a53cb54feecf148c6567279 --- /dev/null +++ b/models/octa-unetplusplus/model_fold2/model_best_f2.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3c62da4ba0f216f3df1f95ffdf3d9b5116467dd85a79ef482cfc48096bfa7e43 +size 379775475 diff --git a/models/octa-unetplusplus/model_fold2/model_last.pt b/models/octa-unetplusplus/model_fold2/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..6292d4401c7f24959a73078674d2dbbb8d5c2ca0 --- /dev/null +++ b/models/octa-unetplusplus/model_fold2/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:510eac56cb470d4aebb8153754773cfe0d10076b679e2a1dd5289869152045d0 +size 379774439 diff --git a/models/octa-unetplusplus/model_fold2_pre/config.json b/models/octa-unetplusplus/model_fold2_pre/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e956173f742b4b1cc4cae8d3f99c83d49837259c --- /dev/null +++ b/models/octa-unetplusplus/model_fold2_pre/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m44_0" + ], + "val_imgs": [ + "m44_0" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m4_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold2_pre", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computeibmgpu-vesta11" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold2_pre/model_best_f2.pt b/models/octa-unetplusplus/model_fold2_pre/model_best_f2.pt new file mode 100644 index 0000000000000000000000000000000000000000..fe219e1d75755c551e9846404c83d0be0b5f2bb6 --- /dev/null +++ b/models/octa-unetplusplus/model_fold2_pre/model_best_f2.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:74c8a318f164b879eb439470efe43c71afb333fbe39525a7b6a1a05e495e02ee +size 379775411 diff --git a/models/octa-unetplusplus/model_fold2_pre/model_last.pt b/models/octa-unetplusplus/model_fold2_pre/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..3b0c6d790a4f07bde8623a416cedc5ad22759571 --- /dev/null +++ b/models/octa-unetplusplus/model_fold2_pre/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:331540c3d5bf765c473c769858541d41f8b2d878ae152825e7a58de63e105b5e +size 379774439 diff --git a/models/octa-unetplusplus/model_fold3/config.json b/models/octa-unetplusplus/model_fold3/config.json new file mode 100644 index 0000000000000000000000000000000000000000..35327bcddce15d184cbe54fbd2b15f3a55f1a04a --- /dev/null +++ b/models/octa-unetplusplus/model_fold3/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m44_1" + ], + "val_imgs": [ + "m44_1" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m4_0", + "m44_0", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold3", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-chiivm0-604" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold3/model_best_f3.pt b/models/octa-unetplusplus/model_fold3/model_best_f3.pt new file mode 100644 index 0000000000000000000000000000000000000000..3f3a29d4067c2f28d0e4d7cb547264fab524dea8 --- /dev/null +++ b/models/octa-unetplusplus/model_fold3/model_best_f3.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9939db83dc1f2a380be16cae8892330548582d8933ef84ef0bba879c671517f2 +size 379775411 diff --git a/models/octa-unetplusplus/model_fold3/model_last.pt b/models/octa-unetplusplus/model_fold3/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..a20cc000aaf5cad97385ee08d62d35ba79bb2c4a --- /dev/null +++ b/models/octa-unetplusplus/model_fold3/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1188d828845a2e0cbcbb1086f21b7cffb337f73f4ff6bbfb39ad6b9c96ca8fb7 +size 379774439 diff --git a/models/octa-unetplusplus/model_fold3_pre/config.json b/models/octa-unetplusplus/model_fold3_pre/config.json new file mode 100644 index 0000000000000000000000000000000000000000..0acd6bb82f420a22a63f7cbe25626107b509be85 --- /dev/null +++ b/models/octa-unetplusplus/model_fold3_pre/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m44_1" + ], + "val_imgs": [ + "m44_1" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m4_0", + "m44_0", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_fold3_pre", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computeibmgpu-vesta10" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold3_pre/model_best_f3.pt b/models/octa-unetplusplus/model_fold3_pre/model_best_f3.pt new file mode 100644 index 0000000000000000000000000000000000000000..b14fced33eb48a09b53fadb132a149cd3809d238 --- /dev/null +++ b/models/octa-unetplusplus/model_fold3_pre/model_best_f3.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f4fc2d9f0837a46aa9bb1c55ac03ee01d32e3f6696a8dce7b5823f926881bb81 +size 379775475 diff --git a/models/octa-unetplusplus/model_fold3_pre/model_last.pt b/models/octa-unetplusplus/model_fold3_pre/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..1556cb7ef5099fdc529340770149d1fb7aecb065 --- /dev/null +++ b/models/octa-unetplusplus/model_fold3_pre/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8f031c5cb6f252826845e25a94c58209877517569c3491b0894b18d900143aa7 +size 379774439 diff --git a/models/octa-unetplusplus/model_fold4/config.json b/models/octa-unetplusplus/model_fold4/config.json new file mode 100644 index 0000000000000000000000000000000000000000..3b25aba94bfbd3bffaf424403f182aca618ae45e --- /dev/null +++ b/models/octa-unetplusplus/model_fold4/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m78_0" + ], + "val_imgs": [ + "m78_0" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m4_0", + "m78_1" + ], + "experiment_name": "superunet_fold4", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-chaiam0-613" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold4/model_best_f4.pt b/models/octa-unetplusplus/model_fold4/model_best_f4.pt new file mode 100644 index 0000000000000000000000000000000000000000..5e7e656823e8d94fcd0762c633c5f3600675777a --- /dev/null +++ b/models/octa-unetplusplus/model_fold4/model_best_f4.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21a9151ddcf3cd46d401d63a35ed9028aab3b18d928ae3f95093278f64b9f163 +size 379775411 diff --git a/models/octa-unetplusplus/model_fold4/model_last.pt b/models/octa-unetplusplus/model_fold4/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..345752727bdb85e9e2fbe3eda5f727a3388d9f61 --- /dev/null +++ b/models/octa-unetplusplus/model_fold4/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bb3b402169c51836bb6b9715e6b50337a356824a4156af95abc0bf0ef756e05d +size 379774439 diff --git a/models/octa-unetplusplus/model_fold4_pre/config.json b/models/octa-unetplusplus/model_fold4_pre/config.json new file mode 100644 index 0000000000000000000000000000000000000000..0207d5fa19307c80cda926fd24ae436b69db831f --- /dev/null +++ b/models/octa-unetplusplus/model_fold4_pre/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m78_0" + ], + "val_imgs": [ + "m78_0" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m4_0", + "m78_1" + ], + "experiment_name": "superunet_fold4_pre", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computegpu-8" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold4_pre/model_best_f4.pt b/models/octa-unetplusplus/model_fold4_pre/model_best_f4.pt new file mode 100644 index 0000000000000000000000000000000000000000..9d1c5813f1d99bf0c2d17fb8ae081369871d6fee --- /dev/null +++ b/models/octa-unetplusplus/model_fold4_pre/model_best_f4.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1cada0e571400b62519fd200f75c9b9477d7fae5ae5961f526e79035160207ec +size 379775411 diff --git a/models/octa-unetplusplus/model_fold4_pre/model_last.pt b/models/octa-unetplusplus/model_fold4_pre/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..f30ab623fcf0c81f4e5bb7cc03e90ea3e8b4d623 --- /dev/null +++ b/models/octa-unetplusplus/model_fold4_pre/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ad0994ddc0689e25ebba355e67076f98a29708af33d7976f0e0d60e1011d50ab +size 379774439 diff --git a/models/octa-unetplusplus/model_fold5/config.json b/models/octa-unetplusplus/model_fold5/config.json new file mode 100644 index 0000000000000000000000000000000000000000..efc53e4e0bd0e9fb5ce87c958a87fc1afbbaa03e --- /dev/null +++ b/models/octa-unetplusplus/model_fold5/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m78_1" + ], + "val_imgs": [ + "m78_1" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m4_0", + "m78_0" + ], + "experiment_name": "superunet_fold5", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-cva0ts0-501" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold5/model_best_f5.pt b/models/octa-unetplusplus/model_fold5/model_best_f5.pt new file mode 100644 index 0000000000000000000000000000000000000000..37b36e7e1d9e822e4e96566e77b22e2989d64044 --- /dev/null +++ b/models/octa-unetplusplus/model_fold5/model_best_f5.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c33d851b985fdf233e9eee68c7eef8d50ff14579836d59208c8bc0cdf988232 +size 379775411 diff --git a/models/octa-unetplusplus/model_fold5/model_last.pt b/models/octa-unetplusplus/model_fold5/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..afe5938af8182de20cdd6749994271bc5d7d61e2 --- /dev/null +++ b/models/octa-unetplusplus/model_fold5/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8312783a19786afdf193726ebda95807e2ad76e24eb0428c520a0902c765f785 +size 379774439 diff --git a/models/octa-unetplusplus/model_fold5_pre/config.json b/models/octa-unetplusplus/model_fold5_pre/config.json new file mode 100644 index 0000000000000000000000000000000000000000..d84c657bb7a3a201edb57ef7badcab062abcb23e --- /dev/null +++ b/models/octa-unetplusplus/model_fold5_pre/config.json @@ -0,0 +1,117 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations_incldoppler", + "test_imgs": [ + "m78_1" + ], + "val_imgs": [ + "m78_1" + ], + "train_on_real": true, + "use_doppler": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m78_0", + "m4_0" + ], + "experiment_name": "superunet_fold5_pre", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 400, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.5, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.2, + 0.2 + ], + "intensity_shift_offsets": [ + -0.2, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "85fc903", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computeibmgpu-vesta6" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_fold5_pre/model_best_f5.pt b/models/octa-unetplusplus/model_fold5_pre/model_best_f5.pt new file mode 100644 index 0000000000000000000000000000000000000000..25551ae08be71e16179200cb2c9f7ca85805776e --- /dev/null +++ b/models/octa-unetplusplus/model_fold5_pre/model_best_f5.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5405dfb44d3fef5d3c9b7a578737cc8461e9076b66eaa49ad6cd275745b9c813 +size 379775411 diff --git a/models/octa-unetplusplus/model_fold5_pre/model_last.pt b/models/octa-unetplusplus/model_fold5_pre/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..f52506c49d7bb30733723146845f584156361025 --- /dev/null +++ b/models/octa-unetplusplus/model_fold5_pre/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7c0a40bc5380f801018953e18b442bebecdaed79ceb5bf77928bc96ac733e476 +size 379774439 diff --git a/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/config.json b/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/config.json new file mode 100644 index 0000000000000000000000000000000000000000..33351da71f095c01b0e528811ca3b8ce17e77c66 --- /dev/null +++ b/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/config.json @@ -0,0 +1,116 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations", + "test_imgs": [ + "m4_0" + ], + "val_imgs": [ + "m4_0" + ], + "train_on_real": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_syn_base_cldice01_moreparams", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 300, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": true, + "iter": 15, + "alpha": 0.1, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.5, + 1.2 + ], + "intensity_shift_offsets": [ + 0.05, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "7a8f7f7", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computegpu-3" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/model_best_f0.pt b/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/model_best_f0.pt new file mode 100644 index 0000000000000000000000000000000000000000..d3c705c8ce34eace6a4aaa05a4db851675aa3621 --- /dev/null +++ b/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/model_best_f0.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cc3f754b23fdeca0759571db2e2bb4a88bca92f01fa3222e85a5a32cd68ff3f4 +size 379775475 diff --git a/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/model_last.pt b/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..1e66279851ae3bf2cdc7678669b9fef44f0ad5ed --- /dev/null +++ b/models/octa-unetplusplus/model_syn_base_cldice01_moreparams/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4f9c74c65c7e5386c89a5b87d6253c94d2af13d2d3b34bd5c62c60b68ee54a47 +size 379774503 diff --git a/models/octa-unetplusplus/model_syn_base_moreparams/config.json b/models/octa-unetplusplus/model_syn_base_moreparams/config.json new file mode 100644 index 0000000000000000000000000000000000000000..35ee8f689a8f99ddf53d0bb7a1b2bd90d23f94af --- /dev/null +++ b/models/octa-unetplusplus/model_syn_base_moreparams/config.json @@ -0,0 +1,116 @@ +{ + "sim_data": "/home/bwittm/share/synthetic_cerebral_octa", + "sim_data_version": "b421", + "annotations": "/home/bwittm/share/manual_annotations", + "test_imgs": [ + "m4_0" + ], + "val_imgs": [ + "m4_0" + ], + "train_on_real": false, + "train_img": [ + "m4_1", + "m44_0", + "m44_1", + "m78_0", + "m78_1" + ], + "experiment_name": "superunet_syn_base_moreparams", + "device": "cuda:0", + "store_ckpt": true, + "epochs": 300, + "epochs_decay": 50, + "val_interval": 1, + "itera_pro_epoch": 50000, + "seed": 0, + "batch_size": 32, + "num_workers": 16, + "sliding_window_overlap": 0.5, + "sliding_window_mode": "constant", + "threshold": 0.5, + "seg": { + "lr": "1e-5", + "channels": [ + 64, + 128, + 256, + 512, + 1024 + ], + "strides": [ + 2, + 2, + 2, + 2 + ], + "activation": "LEAKYRELU", + "normalization": "BATCH", + "dropout": 0.3, + "max_norm_clip": 1.0, + "cldice": false, + "iter": 15, + "alpha": 0.1, + "smooth": 1 + }, + "augmentation": { + "patch_size": [ + 64, + 64, + 64 + ], + "use_augmentation": true, + "p_gaussian_noise": 0.5, + "p_gaussian_smooth": 0.5, + "p_intensity_scale": 0, + "p_intensity_shift": 0, + "p_adjust_contrast": 0, + "p_histogram_shift": 0.5, + "p_rotate": 0.5, + "p_zoom": 0.5, + "p_shear": 0, + "p_flip": [ + 0, + 0.5, + 0.5 + ], + "gaussian_noise_mean": 0.0, + "gaussian_noise_std": 0.01, + "gaussian_smooth_sigma": [ + 0.0, + 0.5 + ], + "intensity_scale_factors": [ + -0.5, + 1.2 + ], + "intensity_shift_offsets": [ + 0.05, + 0.2 + ], + "adjust_contrast_gamma": [ + 0.7, + 1.3 + ], + "control_points": [ + 5, + 10 + ], + "rotation": [ + -90, + 90 + ], + "min_zoom": 1.0, + "max_zoom": 1.2, + "shear_range": [ + 0.1, + 0.1, + 0.1 + ] + }, + "git_commit_hash": "7a8f7f7", + "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ", + "gcc_version": "[GCC 12.3.0]", + "pytorch_version": "1.13.1+cu116", + "host_name": "u20-computeibmgpu-vesta13" +} \ No newline at end of file diff --git a/models/octa-unetplusplus/model_syn_base_moreparams/model_best_f0.pt b/models/octa-unetplusplus/model_syn_base_moreparams/model_best_f0.pt new file mode 100644 index 0000000000000000000000000000000000000000..66091d0aab089361060004653f36beba50ce7c47 --- /dev/null +++ b/models/octa-unetplusplus/model_syn_base_moreparams/model_best_f0.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:49f81d2d0c57169b29f2fd1e7ea0227a5bafa0c5eac7f0611498c0c9b2960d12 +size 379775411 diff --git a/models/octa-unetplusplus/model_syn_base_moreparams/model_last.pt b/models/octa-unetplusplus/model_syn_base_moreparams/model_last.pt new file mode 100644 index 0000000000000000000000000000000000000000..772952b4b5b138acdfbb6fb6817f90e17ad7bc0e --- /dev/null +++ b/models/octa-unetplusplus/model_syn_base_moreparams/model_last.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:670787f0634062eee6b980d9cb6bcf75df702d9e4998e601cafb402152041b68 +size 379774439 diff --git a/models/octa-unetplusplus/source.txt b/models/octa-unetplusplus/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..cff8aa18a3feac4cb346267309dfa7adfa220ee7 --- /dev/null +++ b/models/octa-unetplusplus/source.txt @@ -0,0 +1 @@ +https://huggingface.co/bwittmann/octa-unetplusplus \ No newline at end of file diff --git a/models/unet_plus_plus/.gitattributes b/models/unet_plus_plus/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..c7d9f3332a950355d5a77d85000f05e6f45435ea --- /dev/null +++ b/models/unet_plus_plus/.gitattributes @@ -0,0 +1,34 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unet_plus_plus/config.json b/models/unet_plus_plus/config.json new file mode 100644 index 0000000000000000000000000000000000000000..38ca1a313d348f941a397d528d85082599a49aea --- /dev/null +++ b/models/unet_plus_plus/config.json @@ -0,0 +1,21 @@ +{ + "architectures": [ + "HFUnetPlusPlus" + ], + "auto_map": { + "AutoConfig": "hf_config.UnetConfig", + "AutoModelForImageSegmentation": "hf_model.HFUnetPlusPlus" + }, + "decoder_channels": [ + 1024, + 512, + 256, + 128, + 64 + ], + "encoder_name": "resnet18", + "input_channels": 1, + "num_classes": 16, + "torch_dtype": "float32", + "transformers_version": "4.24.0" +} diff --git a/models/unet_plus_plus/hf_config.py b/models/unet_plus_plus/hf_config.py new file mode 100644 index 0000000000000000000000000000000000000000..30cd00a3e752f6fbd7566f3842414cad750120b5 --- /dev/null +++ b/models/unet_plus_plus/hf_config.py @@ -0,0 +1,17 @@ +from transformers import PretrainedConfig + +class UnetConfig(PretrainedConfig): + + def __init__( + self, + encoder_name: str = "resnet18", + num_classes: int = 16, + input_channels: int = 1, + decoder_channels: tuple = (1024, 512, 256, 128, 64), + **kwargs + ): + self.encoder_name = encoder_name + self.num_classes = num_classes + self.input_channels = input_channels + self.decoder_channels = decoder_channels + super().__init__(**kwargs) diff --git a/models/unet_plus_plus/hf_model.py b/models/unet_plus_plus/hf_model.py new file mode 100644 index 0000000000000000000000000000000000000000..c965c46d0a1ba2d03333aff95ad2a9f5047bbef6 --- /dev/null +++ b/models/unet_plus_plus/hf_model.py @@ -0,0 +1,21 @@ +import segmentation_models_pytorch as smp +from .hf_config import UnetConfig +from transformers import PreTrainedModel + + +class HFUnetPlusPlus(PreTrainedModel): + config_class = UnetConfig + + def __init__(self, config): + super().__init__(config) + + self.model = smp.UnetPlusPlus( + encoder_name=config.encoder_name, + encoder_weights="imagenet", + decoder_channels=config.decoder_channels, + in_channels=config.input_channels, + classes=config.num_classes, + decoder_attention_type="scse") + + def forward(self, tensor): + return self.model(tensor) diff --git a/models/unet_plus_plus/pytorch_model.bin b/models/unet_plus_plus/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..93dbe56eeabed62d12429284029383b10b9a47d7 --- /dev/null +++ b/models/unet_plus_plus/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:11241ff685300ecf0d1314eabc7bf74223d2071bae43b7b3ce7859a01f599efd +size 164204733 diff --git a/models/unet_plus_plus/source.txt b/models/unet_plus_plus/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..cbd0928f5e3e89a4712fb16e509f42fdaac44451 --- /dev/null +++ b/models/unet_plus_plus/source.txt @@ -0,0 +1 @@ +https://huggingface.co/voitl/unet_plus_plus \ No newline at end of file diff --git a/models/unetplusplus-tu-resnet18/.gitattributes b/models/unetplusplus-tu-resnet18/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus-tu-resnet18/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus-tu-resnet18/README.md b/models/unetplusplus-tu-resnet18/README.md new file mode 100644 index 0000000000000000000000000000000000000000..4292cbeccc9b7995906c93ccffeb15c09a8634aa --- /dev/null +++ b/models/unetplusplus-tu-resnet18/README.md @@ -0,0 +1,55 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "tu-resnet18", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_batchnorm": True, + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +[More Information Needed] + +## Dataset +Dataset name: [More Information Needed] + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus-tu-resnet18/config.json b/models/unetplusplus-tu-resnet18/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4ae6ad15b177c3bc4777951423a22941b3b71846 --- /dev/null +++ b/models/unetplusplus-tu-resnet18/config.json @@ -0,0 +1,19 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_use_batchnorm": true, + "encoder_depth": 5, + "encoder_name": "tu-resnet18", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus-tu-resnet18/input-tensor.pth b/models/unetplusplus-tu-resnet18/input-tensor.pth new file mode 100644 index 0000000000000000000000000000000000000000..9919458ed275f4c315fdf51559aa7a15056a563d --- /dev/null +++ b/models/unetplusplus-tu-resnet18/input-tensor.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:43cebe575523ed0d2e9820e994b70e96cff73d577bc5e225ecb2131bf187c006 +size 787637 diff --git a/models/unetplusplus-tu-resnet18/model.safetensors b/models/unetplusplus-tu-resnet18/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..205c38f4672a5b68a23042e383f383211ba08478 --- /dev/null +++ b/models/unetplusplus-tu-resnet18/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a0153c2018eb2b0ba08488203cedce5569fb1012ea9c5512b151dee2ca3b5a89 +size 63962236 diff --git a/models/unetplusplus-tu-resnet18/output-tensor.pth b/models/unetplusplus-tu-resnet18/output-tensor.pth new file mode 100644 index 0000000000000000000000000000000000000000..b7aab82601308e0e44137b5457112f0f5aec6bb0 --- /dev/null +++ b/models/unetplusplus-tu-resnet18/output-tensor.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:67bdf3cae813989e3d1326bc6b2b517fda6906dc935a66af928f49fd9fae2cb0 +size 263354 diff --git a/models/unetplusplus-tu-resnet18/source.txt b/models/unetplusplus-tu-resnet18/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..cab0fb94ba2356bb6553c5fb23ada42127ec03f9 --- /dev/null +++ b/models/unetplusplus-tu-resnet18/source.txt @@ -0,0 +1 @@ +https://huggingface.co/smp-test-models/unetplusplus-tu-resnet18 \ No newline at end of file diff --git a/models/unetplusplus_densenet201/.gitattributes b/models/unetplusplus_densenet201/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_densenet201/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_densenet201/README.md b/models/unetplusplus_densenet201/README.md new file mode 100644 index 0000000000000000000000000000000000000000..e1f18148c8ace6d16f7291686dd51e4ee6a678dc --- /dev/null +++ b/models/unetplusplus_densenet201/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "densenet201", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.23620693385601044, + "test_dataset_iou": 0.18780048191547394, + "test_per_image_accuracy": 0.9674091935157776, + "test_dataset_accuracy": 0.9674091935157776 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_densenet201/config.json b/models/unetplusplus_densenet201/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a8c9ab9a49bd8cdf4c6e4087646440334518bfab --- /dev/null +++ b/models/unetplusplus_densenet201/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "densenet201", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_densenet201/model.safetensors b/models/unetplusplus_densenet201/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..bf68ab069e1b5bad62bdd0092ab945326e193f54 --- /dev/null +++ b/models/unetplusplus_densenet201/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aca5ac1c7a09035fd1a3dbeb4d5f943241582c018cd3311284db6daa53148efb +size 195448012 diff --git a/models/unetplusplus_densenet201/source.txt b/models/unetplusplus_densenet201/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..9e64747d029b539fa3f27b479656e17346d38eb2 --- /dev/null +++ b/models/unetplusplus_densenet201/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_densenet201 \ No newline at end of file diff --git a/models/unetplusplus_dpn131/.gitattributes b/models/unetplusplus_dpn131/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_dpn131/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_dpn131/README.md b/models/unetplusplus_dpn131/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6e1a5646157fdcdc658c352874a56e6de6435f2f --- /dev/null +++ b/models/unetplusplus_dpn131/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "dpn131", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.6082867383956909, + "test_dataset_iou": 0.6192212104797363, + "test_per_image_accuracy": 0.7703589200973511, + "test_dataset_accuracy": 0.7703589200973511 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_dpn131/config.json b/models/unetplusplus_dpn131/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5c42f971d540c2cb91c5e138d1c39f2cb88e30fa --- /dev/null +++ b/models/unetplusplus_dpn131/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "dpn131", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_dpn131/model.safetensors b/models/unetplusplus_dpn131/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2994a1b71f9deed123dc38a5d04a8f65482b1f80 --- /dev/null +++ b/models/unetplusplus_dpn131/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7137fc1d537183ebf3d58b12801bbb9f913e97e96021d1d68bcdda02ac822d83 +size 528626892 diff --git a/models/unetplusplus_dpn131/source.txt b/models/unetplusplus_dpn131/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..d55b2fb24fcf97689f9891645b402c63ec1ebbb5 --- /dev/null +++ b/models/unetplusplus_dpn131/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_dpn131 \ No newline at end of file diff --git a/models/unetplusplus_efficientnet-b7/.gitattributes b/models/unetplusplus_efficientnet-b7/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_efficientnet-b7/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_efficientnet-b7/README.md b/models/unetplusplus_efficientnet-b7/README.md new file mode 100644 index 0000000000000000000000000000000000000000..e21193d59e68807450f05d233923d8bd0d198b3b --- /dev/null +++ b/models/unetplusplus_efficientnet-b7/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "efficientnet-b7", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.29138806462287903, + "test_dataset_iou": 0.24547137320041656, + "test_per_image_accuracy": 0.9696177244186401, + "test_dataset_accuracy": 0.9696177244186401 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_efficientnet-b7/config.json b/models/unetplusplus_efficientnet-b7/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e3a2ed07947a74a133c391c85581cd7227b60bc6 --- /dev/null +++ b/models/unetplusplus_efficientnet-b7/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "efficientnet-b7", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_efficientnet-b7/model.safetensors b/models/unetplusplus_efficientnet-b7/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2b29612c6fe6483397f331c33f4fb18b40c410be --- /dev/null +++ b/models/unetplusplus_efficientnet-b7/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:428ec591a64c25af0b2483e54b4234bfac88df731d9c80356a8158581f65ff1b +size 274050516 diff --git a/models/unetplusplus_efficientnet-b7/source.txt b/models/unetplusplus_efficientnet-b7/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..a4a8bc25403ffd7b8a0edec613f981db50fc99b0 --- /dev/null +++ b/models/unetplusplus_efficientnet-b7/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_efficientnet-b7 \ No newline at end of file diff --git a/models/unetplusplus_inceptionresnetv2/.gitattributes b/models/unetplusplus_inceptionresnetv2/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_inceptionresnetv2/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_inceptionresnetv2/README.md b/models/unetplusplus_inceptionresnetv2/README.md new file mode 100644 index 0000000000000000000000000000000000000000..9bed189a7c6e0f6c65943311e11a7476c3b86f9d --- /dev/null +++ b/models/unetplusplus_inceptionresnetv2/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "inceptionresnetv2", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.19481821358203888, + "test_dataset_iou": 0.16348780691623688, + "test_per_image_accuracy": 0.9655032157897949, + "test_dataset_accuracy": 0.9655031561851501 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_inceptionresnetv2/config.json b/models/unetplusplus_inceptionresnetv2/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4baa1247d386d7549adbe63a97b6af110a652c69 --- /dev/null +++ b/models/unetplusplus_inceptionresnetv2/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "inceptionresnetv2", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_inceptionresnetv2/model.safetensors b/models/unetplusplus_inceptionresnetv2/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..4cdd515bf8d1d2253e3e4e6c9c4a2cdb9ae53b0b --- /dev/null +++ b/models/unetplusplus_inceptionresnetv2/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b7ae421d4e93918f801d38586e094be7e72539cdf6742eb3504f7ce81905f256 +size 284754372 diff --git a/models/unetplusplus_inceptionresnetv2/source.txt b/models/unetplusplus_inceptionresnetv2/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..2a304b5eb6754bee1ff3306481b29ffa9407ddb5 --- /dev/null +++ b/models/unetplusplus_inceptionresnetv2/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_inceptionresnetv2 \ No newline at end of file diff --git a/models/unetplusplus_inceptionv4/.gitattributes b/models/unetplusplus_inceptionv4/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_inceptionv4/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_inceptionv4/README.md b/models/unetplusplus_inceptionv4/README.md new file mode 100644 index 0000000000000000000000000000000000000000..fa61af76ec664adca59e0b066aed151d8b2f17d6 --- /dev/null +++ b/models/unetplusplus_inceptionv4/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "inceptionv4", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.24594824016094208, + "test_dataset_iou": 0.21292220056056976, + "test_per_image_accuracy": 0.9676333069801331, + "test_dataset_accuracy": 0.9676333069801331 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_inceptionv4/config.json b/models/unetplusplus_inceptionv4/config.json new file mode 100644 index 0000000000000000000000000000000000000000..27a89a8ea34c8c9de52919269b35c9faa2f65a4c --- /dev/null +++ b/models/unetplusplus_inceptionv4/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "inceptionv4", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_inceptionv4/model.safetensors b/models/unetplusplus_inceptionv4/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..617231449d3b64d3fe76ed17944529b8c43c5981 --- /dev/null +++ b/models/unetplusplus_inceptionv4/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e140bc036292433f11bccd0cc78b913d6ed4afe5cf11df7e0256270193b2085c +size 237819708 diff --git a/models/unetplusplus_inceptionv4/source.txt b/models/unetplusplus_inceptionv4/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..3cbbc985be3d68bf75bbf6eba8d4ba7ceee0f71a --- /dev/null +++ b/models/unetplusplus_inceptionv4/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_inceptionv4 \ No newline at end of file diff --git a/models/unetplusplus_mobilenet_v2/.gitattributes b/models/unetplusplus_mobilenet_v2/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_mobilenet_v2/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_mobilenet_v2/README.md b/models/unetplusplus_mobilenet_v2/README.md new file mode 100644 index 0000000000000000000000000000000000000000..45aa2053e949a0cf02c7ee51b8db6c2889c81dc9 --- /dev/null +++ b/models/unetplusplus_mobilenet_v2/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "mobilenet_v2", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.2414918690919876, + "test_dataset_iou": 0.23931807279586792, + "test_per_image_accuracy": 0.9677097201347351, + "test_dataset_accuracy": 0.9677096605300903 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_mobilenet_v2/config.json b/models/unetplusplus_mobilenet_v2/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5a7aee238df4e90cb42a27859643444863eb5035 --- /dev/null +++ b/models/unetplusplus_mobilenet_v2/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "mobilenet_v2", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_mobilenet_v2/model.safetensors b/models/unetplusplus_mobilenet_v2/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..07da6eb5be80ded77e37150d9a99847f9032a02d --- /dev/null +++ b/models/unetplusplus_mobilenet_v2/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:99fcab364d43a3383a91162881f762feef0b7a4cfcee6b5fdf53daf877ade273 +size 27492380 diff --git a/models/unetplusplus_mobilenet_v2/source.txt b/models/unetplusplus_mobilenet_v2/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..d4d03b681ada3290f00f8410dbcfe81c98f4621e --- /dev/null +++ b/models/unetplusplus_mobilenet_v2/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_mobilenet_v2 \ No newline at end of file diff --git a/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/.gitattributes b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/README.md b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/README.md new file mode 100644 index 0000000000000000000000000000000000000000..8ebed3aca60e2f52bb749e7abdfcc36ca072da1e --- /dev/null +++ b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/README.md @@ -0,0 +1,4 @@ +--- +license: mit +pipeline_tag: image-segmentation +--- \ No newline at end of file diff --git a/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/epoch=26-val_mIoU=0.91.ckpt b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/epoch=26-val_mIoU=0.91.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..d9209cc2a7b236fcdde218db27416f7780c1ae4c --- /dev/null +++ b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/epoch=26-val_mIoU=0.91.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6896b3e91abf4181ea344efbb13335bb045029eb568452beddb591f35dff66e8 +size 82396153 diff --git a/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/epoch=26-val_mIoU=0.91.onnx b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/epoch=26-val_mIoU=0.91.onnx new file mode 100644 index 0000000000000000000000000000000000000000..c5874f3ee0c9c7493a44feb50bbe5b0855cf40a3 --- /dev/null +++ b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/epoch=26-val_mIoU=0.91.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4f7c9c090bb55fbf1ad90faaa3e47d2b7993f0d1b270841c5a641a1f71b73245 +size 27286606 diff --git a/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/source.txt b/models/unetplusplus_mobilenetv2-encoder_epoch-26-val_mIoU-0.91/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/models/unetplusplus_mobileone_s4/.gitattributes b/models/unetplusplus_mobileone_s4/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_mobileone_s4/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_mobileone_s4/README.md b/models/unetplusplus_mobileone_s4/README.md new file mode 100644 index 0000000000000000000000000000000000000000..f00943cc525cf31f09930daa7a95d7b9bbf96a8f --- /dev/null +++ b/models/unetplusplus_mobileone_s4/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "mobileone_s4", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.22582745552062988, + "test_dataset_iou": 0.1685050129890442, + "test_per_image_accuracy": 0.966795802116394, + "test_dataset_accuracy": 0.966795802116394 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_mobileone_s4/config.json b/models/unetplusplus_mobileone_s4/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7b97e5c598900d0b9e3b897f9f750afce43dbc08 --- /dev/null +++ b/models/unetplusplus_mobileone_s4/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "mobileone_s4", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_mobileone_s4/model.safetensors b/models/unetplusplus_mobileone_s4/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..8c000b6c929b1fecf50b0067c7013ff55fe80696 --- /dev/null +++ b/models/unetplusplus_mobileone_s4/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ae28c8661f6fcb9b5fc3bb6051ee63bf966d1b21e73b30f53945b77b8f445fe1 +size 134225380 diff --git a/models/unetplusplus_mobileone_s4/source.txt b/models/unetplusplus_mobileone_s4/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..8149aadb970cf1b3a353de5aa1b348f8a38afcf3 --- /dev/null +++ b/models/unetplusplus_mobileone_s4/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_mobileone_s4 \ No newline at end of file diff --git a/models/unetplusplus_resnet152/.gitattributes b/models/unetplusplus_resnet152/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_resnet152/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_resnet152/README.md b/models/unetplusplus_resnet152/README.md new file mode 100644 index 0000000000000000000000000000000000000000..b5be4cb9da0395748ed649c3cdc989aa4cd372cc --- /dev/null +++ b/models/unetplusplus_resnet152/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "resnet152", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.1869548112154007, + "test_dataset_iou": 0.1629718691110611, + "test_per_image_accuracy": 0.9661772847175598, + "test_dataset_accuracy": 0.9661772847175598 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_resnet152/config.json b/models/unetplusplus_resnet152/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8e396b215ab14f5a53e1a9351f86ee0c85f6d489 --- /dev/null +++ b/models/unetplusplus_resnet152/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "resnet152", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_resnet152/model.safetensors b/models/unetplusplus_resnet152/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..9a07f21de6e92985a1571a9eaa31b6a2cfd2508f --- /dev/null +++ b/models/unetplusplus_resnet152/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:48f90c8267fddfc13ea2706f96da92c440362796102b7e43826fefba9c95964e +size 335228492 diff --git a/models/unetplusplus_resnet152/source.txt b/models/unetplusplus_resnet152/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..62cae1a1e382d2e8b3908d54015a681f662faa2b --- /dev/null +++ b/models/unetplusplus_resnet152/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_resnet152 \ No newline at end of file diff --git a/models/unetplusplus_resnext101_32x8d/.gitattributes b/models/unetplusplus_resnext101_32x8d/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_resnext101_32x8d/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_resnext101_32x8d/README.md b/models/unetplusplus_resnext101_32x8d/README.md new file mode 100644 index 0000000000000000000000000000000000000000..83da74649cc0a80cb3700382fcb0c82e6b8f3743 --- /dev/null +++ b/models/unetplusplus_resnext101_32x8d/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "resnext101_32x8d", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.547914981842041, + "test_dataset_iou": 0.6563674807548523, + "test_per_image_accuracy": 0.8345138430595398, + "test_dataset_accuracy": 0.8345139026641846 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_resnext101_32x8d/config.json b/models/unetplusplus_resnext101_32x8d/config.json new file mode 100644 index 0000000000000000000000000000000000000000..01ec4b7de337bd3e2282821a631efa83f55cbaa9 --- /dev/null +++ b/models/unetplusplus_resnext101_32x8d/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "resnext101_32x8d", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_resnext101_32x8d/model.safetensors b/models/unetplusplus_resnext101_32x8d/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..ae4fcf6ba95c6a146d1536e8682587bbb001b6bd --- /dev/null +++ b/models/unetplusplus_resnext101_32x8d/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:22c1c559dde43d148eec4330a403c1f90f4764aba0b27317b1fbc29ee917d847 +size 449797340 diff --git a/models/unetplusplus_resnext101_32x8d/source.txt b/models/unetplusplus_resnext101_32x8d/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..c48907f6e088996499e6c8bc6b37d62e174487a1 --- /dev/null +++ b/models/unetplusplus_resnext101_32x8d/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_resnext101_32x8d \ No newline at end of file diff --git a/models/unetplusplus_resnext50_32x4d_depth/.gitattributes b/models/unetplusplus_resnext50_32x4d_depth/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_resnext50_32x4d_depth/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_resnext50_32x4d_depth/nyu-v2-depth-resnext50_32x4d-unetplusplus.pt b/models/unetplusplus_resnext50_32x4d_depth/nyu-v2-depth-resnext50_32x4d-unetplusplus.pt new file mode 100644 index 0000000000000000000000000000000000000000..5321987dc349ad2863b66067ed0752cf44e3e0e5 --- /dev/null +++ b/models/unetplusplus_resnext50_32x4d_depth/nyu-v2-depth-resnext50_32x4d-unetplusplus.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9146cd2c82b8704af3ba3765eecf74827dd3edc75cb3e7af7cf71252ab3de438 +size 194315589 diff --git a/models/unetplusplus_resnext50_32x4d_depth/source.txt b/models/unetplusplus_resnext50_32x4d_depth/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..531397c7bbb08d5e15376ff7c0492863b1f50755 --- /dev/null +++ b/models/unetplusplus_resnext50_32x4d_depth/source.txt @@ -0,0 +1 @@ +https://huggingface.co/amrithanandini/unetplusplus_resnext50_32x4d_depth \ No newline at end of file diff --git a/models/unetplusplus_se_resnext101_32x4d/.gitattributes b/models/unetplusplus_se_resnext101_32x4d/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_se_resnext101_32x4d/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_se_resnext101_32x4d/README.md b/models/unetplusplus_se_resnext101_32x4d/README.md new file mode 100644 index 0000000000000000000000000000000000000000..f0eeca6377ed846d80955a7a65c1ee55713dcc6a --- /dev/null +++ b/models/unetplusplus_se_resnext101_32x4d/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "se_resnext101_32x4d", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.1955837905406952, + "test_dataset_iou": 0.14369702339172363, + "test_per_image_accuracy": 0.9666634202003479, + "test_dataset_accuracy": 0.9666634202003479 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_se_resnext101_32x4d/config.json b/models/unetplusplus_se_resnext101_32x4d/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f172798aa0a0cf106c3afd448921bae5d4cf60ff --- /dev/null +++ b/models/unetplusplus_se_resnext101_32x4d/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "se_resnext101_32x4d", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_se_resnext101_32x4d/model.safetensors b/models/unetplusplus_se_resnext101_32x4d/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2654530844f7c8be393f069cc58da52d27730bda --- /dev/null +++ b/models/unetplusplus_se_resnext101_32x4d/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:09b3ff0d736f250713618b21a722d8818d224a881ad2ca21ace898a265fec934 +size 290207908 diff --git a/models/unetplusplus_se_resnext101_32x4d/source.txt b/models/unetplusplus_se_resnext101_32x4d/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..082e9eb183c1b36fc60f30267212fbcbed24ab37 --- /dev/null +++ b/models/unetplusplus_se_resnext101_32x4d/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_se_resnext101_32x4d \ No newline at end of file diff --git a/models/unetplusplus_timm-efficientnet-b7/.gitattributes b/models/unetplusplus_timm-efficientnet-b7/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b7/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_timm-efficientnet-b7/README.md b/models/unetplusplus_timm-efficientnet-b7/README.md new file mode 100644 index 0000000000000000000000000000000000000000..8f3740d859a006a5556cb48498c266f428869c26 --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b7/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "timm-efficientnet-b7", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.23284436762332916, + "test_dataset_iou": 0.2990565001964569, + "test_per_image_accuracy": 0.971991777420044, + "test_dataset_accuracy": 0.971991777420044 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_timm-efficientnet-b7/config.json b/models/unetplusplus_timm-efficientnet-b7/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e2843ba8ba3c22b5994c8b51d57035e7dca87108 --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b7/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "timm-efficientnet-b7", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_timm-efficientnet-b7/model.safetensors b/models/unetplusplus_timm-efficientnet-b7/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..b0390ee3b1cc40df36f3be9ef5c57fa8f1c2c577 --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b7/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f850107e09c2e0e441e9edee9130d36f4c596456387d1cd989309e7b083ad6f3 +size 274049676 diff --git a/models/unetplusplus_timm-efficientnet-b7/source.txt b/models/unetplusplus_timm-efficientnet-b7/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..338a7b9865e276851438a121f32f6bf9f2ededd9 --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b7/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_timm-efficientnet-b7 \ No newline at end of file diff --git a/models/unetplusplus_timm-efficientnet-b8/.gitattributes b/models/unetplusplus_timm-efficientnet-b8/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b8/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_timm-efficientnet-b8/README.md b/models/unetplusplus_timm-efficientnet-b8/README.md new file mode 100644 index 0000000000000000000000000000000000000000..d978abf2482b73832525f2a9e0d3f6ee6e226bb3 --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b8/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "timm-efficientnet-b8", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.21585744619369507, + "test_dataset_iou": 0.2068282663822174, + "test_per_image_accuracy": 0.9685878753662109, + "test_dataset_accuracy": 0.9685878753662109 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_timm-efficientnet-b8/config.json b/models/unetplusplus_timm-efficientnet-b8/config.json new file mode 100644 index 0000000000000000000000000000000000000000..78f2d4daab053944b4195596b605aec30e030617 --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b8/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "timm-efficientnet-b8", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_timm-efficientnet-b8/model.safetensors b/models/unetplusplus_timm-efficientnet-b8/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..7ddbfc84646e83e9383c4fe16d8fe89abb9f72d8 --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b8/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c73951bffb07a4ef1f5a09b071b01599d883e9667a9ba65f5eeb02b1da54477f +size 359477868 diff --git a/models/unetplusplus_timm-efficientnet-b8/source.txt b/models/unetplusplus_timm-efficientnet-b8/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..38b583d13a3e4900da61943200f91a25b41d2ead --- /dev/null +++ b/models/unetplusplus_timm-efficientnet-b8/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_timm-efficientnet-b8 \ No newline at end of file diff --git a/models/unetplusplus_vgg19/.gitattributes b/models/unetplusplus_vgg19/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_vgg19/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_vgg19/README.md b/models/unetplusplus_vgg19/README.md new file mode 100644 index 0000000000000000000000000000000000000000..cf61379860fe981849bf25daaf79a96c89c7710e --- /dev/null +++ b/models/unetplusplus_vgg19/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "vgg19", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.18644322454929352, + "test_dataset_iou": 0.17293575406074524, + "test_per_image_accuracy": 0.9658230543136597, + "test_dataset_accuracy": 0.9658230543136597 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_vgg19/config.json b/models/unetplusplus_vgg19/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b50f27d826e96d213516d196a997ec2b60a42fb2 --- /dev/null +++ b/models/unetplusplus_vgg19/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "vgg19", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_vgg19/model.safetensors b/models/unetplusplus_vgg19/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..c77bd9c3c121ffc96a3d51d7a98607c215577df4 --- /dev/null +++ b/models/unetplusplus_vgg19/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:66bdf3c3596eb1e6318a85328f2dec21d79c76c6467602225414d69613b8e16a +size 178685404 diff --git a/models/unetplusplus_vgg19/source.txt b/models/unetplusplus_vgg19/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..b2d32963aec90c1d8f934775e0df8dbf2aff7515 --- /dev/null +++ b/models/unetplusplus_vgg19/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_vgg19 \ No newline at end of file diff --git a/models/unetplusplus_xception/.gitattributes b/models/unetplusplus_xception/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..a6344aac8c09253b3b630fb776ae94478aa0275b --- /dev/null +++ b/models/unetplusplus_xception/.gitattributes @@ -0,0 +1,35 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text diff --git a/models/unetplusplus_xception/README.md b/models/unetplusplus_xception/README.md new file mode 100644 index 0000000000000000000000000000000000000000..a4a4b5404076fb26347a8e7f8435ae8ba63a9de9 --- /dev/null +++ b/models/unetplusplus_xception/README.md @@ -0,0 +1,65 @@ +--- +library_name: segmentation-models-pytorch +license: mit +pipeline_tag: image-segmentation +tags: +- model_hub_mixin +- pytorch_model_hub_mixin +- segmentation-models-pytorch +- semantic-segmentation +- pytorch +languages: +- python +--- +# UnetPlusPlus Model Card + +Table of Contents: +- [Load trained model](#load-trained-model) +- [Model init parameters](#model-init-parameters) +- [Model metrics](#model-metrics) +- [Dataset](#dataset) + +## Load trained model +```python +import segmentation_models_pytorch as smp + +model = smp.from_pretrained("") +``` + +## Model init parameters +```python +model_init_params = { + "encoder_name": "xception", + "encoder_depth": 5, + "encoder_weights": "imagenet", + "decoder_use_norm": "batchnorm", + "decoder_channels": (256, 128, 64, 32, 16), + "decoder_attention_type": None, + "decoder_interpolation": "nearest", + "in_channels": 3, + "classes": 1, + "activation": None, + "aux_params": None +} +``` + +## Model metrics +```json +[ + { + "test_per_image_iou": 0.21848143637180328, + "test_dataset_iou": 0.12569858133792877, + "test_per_image_accuracy": 0.9661090970039368, + "test_dataset_accuracy": 0.966109037399292 + } +] +``` + +## Dataset +Dataset name: VIP + +## More Information +- Library: https://github.com/qubvel/segmentation_models.pytorch +- Docs: https://smp.readthedocs.io/en/latest/ + +This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) \ No newline at end of file diff --git a/models/unetplusplus_xception/config.json b/models/unetplusplus_xception/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a8fd760c1e57e19114a31be3595e23d595448238 --- /dev/null +++ b/models/unetplusplus_xception/config.json @@ -0,0 +1,20 @@ +{ + "_model_class": "UnetPlusPlus", + "activation": null, + "aux_params": null, + "classes": 1, + "decoder_attention_type": null, + "decoder_channels": [ + 256, + 128, + 64, + 32, + 16 + ], + "decoder_interpolation": "nearest", + "decoder_use_norm": "batchnorm", + "encoder_depth": 5, + "encoder_name": "xception", + "encoder_weights": "imagenet", + "in_channels": 3 +} \ No newline at end of file diff --git a/models/unetplusplus_xception/model.safetensors b/models/unetplusplus_xception/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f184a74573e4f51aab20112c65cfb1e3b0add6af --- /dev/null +++ b/models/unetplusplus_xception/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:95dd5ce34152879ae0e4fb36b44463ac7829618665c01081a3bec59491a1bd12 +size 136306964 diff --git a/models/unetplusplus_xception/source.txt b/models/unetplusplus_xception/source.txt new file mode 100644 index 0000000000000000000000000000000000000000..d2d7a9c9b179c83bb7bfd811e773177870a4c0af --- /dev/null +++ b/models/unetplusplus_xception/source.txt @@ -0,0 +1 @@ +https://huggingface.co/Diamantis99/unetplusplus_xception \ No newline at end of file