mr3z4 / semanticsegmentation Goto Github PK

This is a repository to train semantic segmetation models. It will be improved over time.
Explore the docs »

View Demo · Report Bug · Request Feature

This project started as my master thesis. I will keep improving it as long as i can.

• Python 3.7
• PyTorch 1.2.0
• torchvision 0.4.0

To get a local copy up and running follow these simple steps.

First install the requierments as followed.

pip install -r requirments.txt

1. Clone the repo

   git clone https://github.com/MR3z4/SemanticSegmentation.git

2. Run the training code

   python main.py

It will be completed over time.

• Multi GPU support
• Add RMI loss
• Add Mixup option for training
• Add Mixup Without Hesitation for training with mixup
• Add AdaBelief optimizer option for training
• Add CE2P Network(with normal BatchNorm) for training.
• Add InPlace Active BatchNorm for CE2P.
• Add SCHP completely.
• Add Edge Loss
• Add Vision Transformers
• Add MixMatch option for training
• Add FixMatch option for training
• Distibuted support

See the open issues for a list of proposed features (and known issues).

Contributions are what make the open source community such an amazing place to be learn, inspire, and create. Any contributions you make are greatly appreciated.

1. Fork the Project
2. Create your Feature Branch (git checkout -b feature/AmazingFeature)
3. Commit your Changes (git commit -m 'Add some AmazingFeature')
4. Push to the Branch (git push origin feature/AmazingFeature)
5. Open a Pull Request

Distributed under the MIT License. See LICENSE for more information.

M.Mohammadzade - @realMReza - [email protected]

Project Link: https://github.com/MR3z4/SemanticSegmentation

• Peike Li, Yunqiu Xu, Yunchao Wei, Yi Yang. "Self-Correction for Human Parsing" IEEE Transactions on Pattern Analysis and Machine Intelligence 2020, arXiv:1910.09777, Project Code
• Hongyi Zhang, Moustapha Cisse, Yann N. Dauphin, David Lopez-Paz. "mixup: Beyond Empirical Risk Minimization." arXiv preprint arXiv:1710.09412. arXiv:1710.09412
• Hao Yu, Huanyu Wang, Jianxin Wu. "Mixup Without Hesitation" arXiv preprint arXiv:2101.04342. arXiv:2101.04342
• Shuai Zhao, Yang Wang, Zheng Yang, Deng Cai. "Region Mutual Information Loss for Semantic Segmentation", NeurIPS 2019, arXiv:1910.12037, Project Code
• Juntang Zhuang, Tommy Tang, Yifan Ding , Sekhar Tatikonda, Nicha Dvornek, Xenophon Papademetris, James S. Duncan. "AdaBelief Optimizer: fast as Adam, generalizes as good as SGD, and sufficiently stable to train GANs." arXiv:2010.07468, Project Code