Adding segmentation models based model sessions

worker.py

@@ -21,6 +21,7 @@ from web_tool.ModelSessionKerasExample import KerasDenseFineTune
 from web_tool.ModelSessionPytorchSolar import SolarFineTuning
 from web_tool.ModelSessionPyTorchExample import TorchFineTuning
 from web_tool.ModelSessionRandomForest import ModelSessionRandomForest
+from web_tool.ModelSessionPyTorchSegmentationModel import TorchSegmentationModel
 from web_tool.Utils import setup_logging, serialize, deserialize
 
 from web_tool.Models import load_models
@@ -93,6 +94,8 @@ def main():
         model = SolarFineTuning(args.gpu_id, **model_configs[args.model_key])
     elif model_type == "random_forest":
         model = ModelSessionRandomForest(**model_configs[args.model_key])
+    elif model_type == "pytorch_segmodel":
+        model = TorchSegmentationModel(args.gpu_id, **model_configs[args.model_key])
     else:
         raise NotImplementedError("The given model type is not implemented yet.")