- Try out other segmentation networks (see e.g. https://huggingface.co/tasks/image-segmentation) - Refactor the script to better separate parts that need additional features from resnet/mobilnet and parts that only need the segmentation - Try to have an API/Wrapper to add more networks - Decide which networks should be officially supported