U-Net segmentation algorithm with options of pretrained resnet34 and resnet50 encoders. All of the project dockerized with gpu suppport on anaconda environment with multiple loss support..
python docker dockerfile docker-container docker-image image-processing pytorch resnet image-segmentation preprocessing unet resnet34 pytorch-cnn resnet50 unet-pytorch unet-image-segmentation leaf-disease-classification multiple-loss-function
-
Updated
Sep 12, 2024 - Python