Image Semantic Segmentation.
- Segmentation_models (https://segmentation-models.readthedocs.io/en/latest/)
- Tensorflow
- Matplotlib
https://www.kaggle.com/datasets/residentmario/ade20k-outdoors
Trained UNet model. Image input shape - 256x256.
Labels:
- unknown
- building
- sky
- tree
- road
- human
- water
- streetlight
- mountatin
- car
- bicycle
- animal
- bridge
- ground
- grass
- pole
- trafficlight
- motorbike
Architecture:
You can download the best model here - https://drive.google.com/drive/folders/14p1CXtTpotvRcRYym6AmzA8_zfZI4Trq?usp=sharing
- Download the model and place it in the folder with the project.
- cd to the folder
- python3 nn_interpretation.py