waterljwant / Ssc
Semantic Scene Completion
Stars: ✭ 29
Programming Languages
python
139335 projects - #7 most used programming language
Projects that are alternatives of or similar to Ssc
Kiu Net Pytorch
Official Pytorch Code of KiU-Net for Image Segmentation - MICCAI 2020 (Oral)
Stars: ✭ 134 (+362.07%)
Mutual labels: deep-neural-networks, semantic-segmentation
Medicaldetectiontoolkit
The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.
Stars: ✭ 917 (+3062.07%)
Mutual labels: deep-neural-networks, semantic-segmentation
Refinenet
RefineNet: Multi-Path Refinement Networks for High-Resolution Semantic Segmentation
Stars: ✭ 543 (+1772.41%)
Mutual labels: deep-neural-networks, semantic-segmentation
Crfasrnn pytorch
CRF-RNN PyTorch version http://crfasrnn.torr.vision
Stars: ✭ 102 (+251.72%)
Mutual labels: deep-neural-networks, semantic-segmentation
Keras Unet
Helper package with multiple U-Net implementations in Keras as well as useful utility tools helpful when working with image semantic segmentation tasks. This library and underlying tools come from multiple projects I performed working on semantic segmentation tasks
Stars: ✭ 196 (+575.86%)
Mutual labels: deep-neural-networks, semantic-segmentation
Cascaded Fcn
Source code for the MICCAI 2016 Paper "Automatic Liver and Lesion Segmentation in CT Using Cascaded Fully Convolutional NeuralNetworks and 3D Conditional Random Fields"
Stars: ✭ 296 (+920.69%)
Mutual labels: deep-neural-networks, semantic-segmentation
Tusimple Duc
Understanding Convolution for Semantic Segmentation
Stars: ✭ 567 (+1855.17%)
Mutual labels: deep-neural-networks, semantic-segmentation
All Classifiers 2019
A collection of computer vision projects for Acute Lymphoblastic Leukemia classification/early detection.
Stars: ✭ 22 (-24.14%)
Mutual labels: deep-neural-networks
Face Parsing.pytorch
Using modified BiSeNet for face parsing in PyTorch
Stars: ✭ 838 (+2789.66%)
Mutual labels: semantic-segmentation
Deep Embedded Memory Networks
https://arxiv.org/abs/1707.00836
Stars: ✭ 19 (-34.48%)
Mutual labels: deep-neural-networks
Semantic Segmentation Editor
Web labeling tool for bitmap images and point clouds
Stars: ✭ 894 (+2982.76%)
Mutual labels: semantic-segmentation
Servenet
Service Classification based on Service Description
Stars: ✭ 21 (-27.59%)
Mutual labels: deep-neural-networks
3d Semantic Segmentation For Scene Parsing
A new approach for the real time 3D semantic segmentation based on feature abstract and deep learning method
Stars: ✭ 13 (-55.17%)
Mutual labels: semantic-segmentation
Tf Keras Surgeon
Pruning and other network surgery for trained TF.Keras models.
Stars: ✭ 25 (-13.79%)
Mutual labels: deep-neural-networks
Onnx Tensorflow
Tensorflow Backend for ONNX
Stars: ✭ 846 (+2817.24%)
Mutual labels: deep-neural-networks
Efficientdet Pytorch
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
Stars: ✭ 906 (+3024.14%)
Mutual labels: semantic-segmentation
Tabnet
PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf
Stars: ✭ 882 (+2941.38%)
Mutual labels: deep-neural-networks
Image classification cifar 10
Image Classification on CIFAR-10 Dataset using Multi Layer Perceptrons in Python from Scratch.
Stars: ✭ 18 (-37.93%)
Mutual labels: deep-neural-networks
Zoneout Tensorflow
An implementation of zoneout regularizer on LSTM-RNN by Tensorflow
Stars: ✭ 23 (-20.69%)
Mutual labels: deep-neural-networks
Kaggle Carvana Image Masking Challenge
Top 15% ranked solution to the Carvana Image Masking Challenge on Kaggle
Stars: ✭ 13 (-55.17%)
Mutual labels: semantic-segmentation
Semantic Scene Completion
This repo contains code for the following papers
- AICNet (CVPR2020): Anisotropic Convolutional Networks for 3D Semantic Scene Completion
- DDRNet (CVPR2019): RGBD Based Dimensional Decomposition Residual Network for 3D Semantic Scene Completion
- PALNet (RAL2019): Depth Based Semantic Scene Completion with Position Importance Aware Loss
Contents
Installation
Environment
- Ubuntu 16.04
- python 3.6
- CUDA 10.1
Requirements:
- pytorch≥1.4.0
- torch_scatter
- imageio
- scipy
- scikit-learn
- tqdm
You can install the requirements by running pip install -r requirements.txt
.
If you use other versions of PyTorch or CUDA, be sure to select the corresponding version of torch_scatter.
Data Preparation
Download dataset
The raw data can be found in SSCNet.
The repackaged data can be downloaded via Google Drive or BaiduYun(Access code:lpmk).
The repackaged data includes:
rgb_tensor = npz_file['rgb'] # pytorch tensor of color image
depth_tensor = npz_file['depth'] # pytorch tensor of depth
tsdf_hr = npz_file['tsdf_hr'] # flipped TSDF, (240, 144, 240)
tsdf_lr = npz_file['tsdf_lr'] # flipped TSDF, ( 60, 36, 60)
target_hr = npz_file['target_hr'] # ground truth, (240, 144, 240)
target_lr = npz_file['target_lr'] # ground truth, ( 60, 36, 60)
position = npz_file['position'] # 2D-3D projection mapping index
Train and Test
config.py
Configure the data path in'train': '/path/to/your/training/data'
'val': '/path/to/your/testing/data'
Train
Edit the training script run_SSC_train.sh, then run
bash run_SSC_train.sh
Test
Edit the testing script run_SSC_test.sh, then run
bash run_SSC_test.sh
Visualization and Evaluation
comging soon
Citation
If you find this work useful in your research, please cite our paper(s):
@inproceedings{Li2020aicnet,
author = {Jie Li, Kai Han, Peng Wang, Yu Liu, and Xia Yuan},
title = {Anisotropic Convolutional Networks for 3D Semantic Scene Completion},
booktitle = {Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
year = {2020},
}
@InProceedings{Li2019ddr,
author = {Li, Jie and Liu, Yu and Gong, Dong and Shi, Qinfeng and Yuan, Xia and Zhao, Chunxia and Reid, Ian},
title = {RGBD Based Dimensional Decomposition Residual Network for 3D Semantic Scene Completion},
booktitle = {Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
month = {June},
pages = {7693--7702},
year = {2019}
}
@article{li2019palnet,
title={Depth Based Semantic Scene Completion With Position Importance Aware Loss},
author={Li, Jie and Liu, Yu and Yuan, Xia and Zhao, Chunxia and Siegwart, Roland and Reid, Ian and Cadena, Cesar},
journal={IEEE Robotics and Automation Letters},
volume={5},
number={1},
pages={219--226},
year={2019},
publisher={IEEE}
}
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].