LibtorchSegmentation
A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.
File Explorer
Download Latest Version (.zip)- cmake_buid_here.txt
- training tricks.md
- DeepLab.cpp
- DeepLab.h
- DeepLabDecoder.cpp
- DeepLabDecoder.h
- FPN.cpp
- FPN.h
- FPNDecoder.cpp
- FPNDecoder.h
- LinkNet.cpp
- LinkNet.h
- LinknetDecoder.cpp
- LinknetDecoder.h
- PAN.cpp
- PAN.h
- PANDecoder.cpp
- PANDecoder.h
- PSPNet.cpp
- PSPNet.h
- PSPNetDecoder.cpp
- PSPNetDecoder.h
- UNet.cpp
- UNet.h
- UNetDecoder.cpp
- UNetDecoder.h
- ResNet.cpp
- ResNet.h
- VGG.cpp
- VGG.h
- _dirent.h
- Augmentations.cpp
- Augmentations.h
- InterFace.h
- json.hpp
- loss.h
- readfile.h
- util.cpp
- util.h
- SegDataset.cpp
- SegDataset.h
- Segmentor.cpp
- Segmentor.h
- CMakeLists.txt
- resnet34.cpp
- train.cpp
- 2007_000027.jpg
- 2007_000027.json
- 2007_000170.jpg
- 2007_000170.json
- 2007_000272.jpg
- 2007_000272.json
- 2007_000323.jpg
- 2007_000346.jpg
- 2007_000346.json
- 2007_000423.jpg
- 2007_000423.json
- 2007_000664.jpg
- 2007_000664.json
- 2007_000733.jpg
- 2007_000733.json
- 2007_000762.jpg
- 2007_000783.jpg
- 2007_000783.json
- 2007_000799.jpg
- 2007_000799.json
- 2007_000807.jpg
- 2007_000807.json
- 2007_000836.jpg
- 2007_000836.json
- 2007_000847.jpg
- 2007_000847.json
- 2007_000999.jpg
- 2007_000999.json
- 2007_001185.jpg
- 2007_001185.json
- 2007_001408.jpg
- 2007_001408.json
- 2007_001430.jpg
- 2007_001430.json
- 2007_001526.jpg
- 2007_001558.jpg
- 2007_001583.jpg
- 2007_001585.jpg
- 2007_001585.json
- 2007_001627.jpg
- 2007_001627.json
- 2007_001630.jpg
- 2007_001630.json
- 2007_001717.jpg
- 2007_001717.json
- 2007_002293.jpg
- 2007_002293.json
- 2007_002668.jpg
- 2007_002728.jpg
- 2007_002728.json
- 2007_002824.jpg
- 2007_002824.json
- 2007_003106.jpg
- 2007_003106.json
- 2007_003118.jpg
- 2007_003188.jpg
- 2007_003188.json
- 2007_003189.jpg
- 2007_003189.json
- 2007_003191.jpg
- 2007_003191.json
- 2007_003205.jpg
- 2007_003205.json
- 2007_003329.jpg
- 2007_003529.jpg
- 2007_003530.jpg
- 2007_003530.json
- 2007_003541.jpg
- 2007_003541.json
- 2007_003742.jpg
- 2007_003742.json
- 2007_003745.jpg
- 2007_003745.json
- 2007_003747.jpg
- 2007_003747.json
- 2007_004000.jpg
- 2007_004000.json
- 2007_004049.jpg
- 2007_004133.jpg
- 2007_004133.json
- 2007_004197.jpg
- 2007_004197.json
- 2007_004291.jpg
- 2007_004392.jpg
- 2007_004397.jpg
- 2007_004476.jpg
- 2007_004476.json
- 2007_004712.jpg
- 2007_004712.json
- 2007_004831.jpg
- 2007_004902.jpg
- 2007_005019.jpg
- 2007_005086.jpg
- 2007_005086.json
- 2007_005124.jpg
- 2007_005124.json
- input weights here.txt
- .gitignore
- CMakeLists.txt
- Config.cmake.in
- LICENSE
- prediction.jpg
- README-Chinese.md
- README.md
- segmentation.pc.in
- trace.py
# Installation Guide
1. Get the code
git clone https://github.com/AllentDan/LibtorchSegmentation
Downloads the entire project code from GitHub to your computer.
cd LibtorchSegmentation
Moves into the project folder you just downloaded.
2. CMake
Medium RecommendedPrerequisites
cmake -DBUILD_SHARED=TRUE ..
Type this command into your terminal and run it.
cmake ..
Analyzes the source code and generates build configuration files (must be run inside the build folder).
Check that an executable was created inside the build folder, then run it directly (e.g. ./build/app_name).
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)
