Image Semantic Segmentation Example
- Ubuntu 18.04
- NVIDIA RTX 3090
- CUDA 11.1
- CuDNN 8.1.0
- python 3.8.12
- pytorch 1.9.1
- opencv-python 4.5.4.58
- albumentations 1.1.0
- numpy 1.19.2
- tqdm 4.62.3
- matplotlib 3.2.2
imassege
├──dataset/
| ├── your-own-dataset/
├──dcnn_module/
| ├── dataset_module/
| | ├── interface/
| | | ├── abstract_dataset.py
| | | ├── blur_detection.py
| | | ├── tgs_salt.py
| | ├── connection.py
| | ├── dataset.py
| ├── neural_network/
| | ├── deeplab/
| | | ├── backbone/
| | | | ├── __init__.py
| | | | ├── drn.py
| | | | ├── mobilenet.py
| | | | ├── resnet.py
| | | | ├── xception.py
| | | ├── decoder/
| | | | ├── __init__.py
| | | | ├── deeplab_decoder.py
| | | | ├── unet_decoder.py
| | | ├── sync_batchnorm/
| | | | ├── __init__.py
| | | | ├── batchnorm.py
| | | | ├── comm.py
| | | | ├── replicate.py
| | | | ├── unittest.py
| | | ├── __init__.py
| | | ├── aspp.py
| | | ├── deeplab.py
| | ├── unet.py
| | ├── plainUnet.py
| | ├── densePlainUnet.py
| | ├── mini_unet.py
| ├── utils/
| | ├── metrics_numpy.py
| | ├── metrics_pytorch.py
| | ├── preprocessing.py
| ├── config.py
├──output/
| ├── predictions/
├──train.py
├──predict.py
├──README.md