https://github.com/koushik-elite/image-segmentation
Image Segmentation Sample using pytorch
https://github.com/koushik-elite/image-segmentation
computer-vision deep-learning deep-neural-networks image-recognition image-segmentation object-detection python3 pytorch rcnn
Last synced: 2 months ago
JSON representation
Image Segmentation Sample using pytorch
- Host: GitHub
- URL: https://github.com/koushik-elite/image-segmentation
- Owner: koushik-elite
- Created: 2019-09-08T17:52:48.000Z (over 5 years ago)
- Default Branch: master
- Last Pushed: 2019-09-16T11:33:23.000Z (over 5 years ago)
- Last Synced: 2025-01-22T01:36:35.072Z (4 months ago)
- Topics: computer-vision, deep-learning, deep-neural-networks, image-recognition, image-segmentation, object-detection, python3, pytorch, rcnn
- Language: Jupyter Notebook
- Homepage:
- Size: 2.39 MB
- Stars: 1
- Watchers: 3
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Image Segmentation
Image Segmentation using pytorch
Setup Information
### maskrcnn_benchmark and coco api dependencies
pip install ninja yacs cython matplotlib tqdm opencv-python
### follow PyTorch installation in https://pytorch.org/get-started/locally/
### we give the instructions for CUDA 9.0
pip install -c pytorch pytorch-nightly torchvision cudatoolkit=9.0
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install
cd ../../
### install apex
rm -rf apex
git clone https://github.com/NVIDIA/apex.git
cd apex
git pull
python setup.py install --cuda_ext --cpp_ext
cd ../
### install PyTorch Detection
git clone https://github.com/facebookresearch/maskrcnn-benchmark.git
cd maskrcnn-benchmark
### the following will install the lib with
### symbolic links, so that you can modify
### the files if you want and won't need to
### re-build it
python setup.py build develop
Reference
https://github.com/facebookresearch/maskrcnn-benchmark