Mask Predictor

Semantic segmentation of a scene and output corresponding class' depth map

Requirements

python3 torchvision pickle numpy opencv-python scikit-image matplotlib detectron2

https://github.com/usmanzahidi/MaskPredictor.git

usage: main.py [-r PATH] [-d PATH]
-r rgb   image folder path
-d depth image folder path

python main.py -r ./images/rgb/file_name.png -d ./images/depth/file_name.png

Name		Name	Last commit message	Last commit date
Latest commit History 29 Commits
data		data
images		images
visualizer		visualizer
README.md		README.md
main.py		main.py
masks_predictor.py		masks_predictor.py
requirements.txt		requirements.txt