# uyolo-segmentation **Repository Path**: uyolo-cn/uyolo-segmentation ## Basic Information - **Project Name**: uyolo-segmentation - **Description**: No description available - **Primary Language**: Python - **License**: Apache-2.0 - **Default Branch**: main - **Homepage**: None - **GVP Project**: No ## Statistics - **Stars**: 0 - **Forks**: 0 - **Created**: 2023-03-04 - **Last Updated**: 2023-08-13 ## Categories & Tags **Categories**: Uncategorized **Tags**: None ## README #
uyolo-segmentation

SOTA Real Time Semantic Segmentation Models in PyTorch

Open In Colab
# Introduction `uyolo-segmentation` is an open source semantic segmentation toolbox based on [`PyTorch Lightning`](https://github.com/Lightning-AI/lightning). # Getting Started ## Installation ### 1. Install from Source Clone the `uyolo-segmentation` repo from Github. ```shell git clone https://github.com/uyolo1314/uyolo-segmentation ``` Run the following command, install `uyolo-segmentation` from source. If you make modification to `uyolo-segmentation/uyoloseg`, it will be efficient without reinstallation. ```shell cd uyolo-segmentation pip install -r requirements.txt pip install -v -e . ``` ## How to Train `uyolo-segmentation` is now using pytorch lightning for training. For both single-GPU or multiple-GPUs, run: ```shell config_file=configs/custom.yaml python tools/train.py --cfg ${config_file} ``` ## Deployment - [ ] TensorRT - [ ] OpenVINO - [ ] NCNN - [ ] Ascend # License `uyolo-segmentation` is released under the [Apache 2.0 license](./LICENSE). # Acknowledgement - https://github.com/PaddlePaddle/PaddleSeg - https://github.com/open-mmlab/mmsegmentation - https://github.com/sithu31296/semantic-segmentation - https://github.com/RangiLyu/nanodet - https://github.com/Lightning-AI/lightning