├── .gitignore ├── LICENSE ├── README.md ├── args.py ├── cache ├── class_weights_nyu40.txt ├── class_weights_scannet20.txt ├── test.txt └── train.txt ├── data ├── README.md ├── __init__.py ├── scannet.py └── utils.py ├── images └── scannet_demo.png ├── inference.py ├── main.py ├── metric ├── __init__.py ├── confusionmatrix.py ├── iou.py └── metric.py ├── models └── enet.py ├── prepare_data ├── README.md ├── SensorData.py ├── prepare_2d_data.py ├── reader.py └── util.py ├── requirements.txt ├── save ├── ENetDepth-scannet20 └── ENetDepth-scannet20_summary.txt ├── test.py ├── train.py ├── transforms.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/README.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/args.py -------------------------------------------------------------------------------- /cache/class_weights_nyu40.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/cache/class_weights_nyu40.txt -------------------------------------------------------------------------------- /cache/class_weights_scannet20.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/cache/class_weights_scannet20.txt -------------------------------------------------------------------------------- /cache/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/cache/test.txt -------------------------------------------------------------------------------- /cache/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/cache/train.txt -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/data/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/data/scannet.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/data/utils.py -------------------------------------------------------------------------------- /images/scannet_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/images/scannet_demo.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/inference.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/main.py -------------------------------------------------------------------------------- /metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/metric/__init__.py -------------------------------------------------------------------------------- /metric/confusionmatrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/metric/confusionmatrix.py -------------------------------------------------------------------------------- /metric/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/metric/iou.py -------------------------------------------------------------------------------- /metric/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/metric/metric.py -------------------------------------------------------------------------------- /models/enet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/models/enet.py -------------------------------------------------------------------------------- /prepare_data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/prepare_data/README.md -------------------------------------------------------------------------------- /prepare_data/SensorData.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/prepare_data/SensorData.py -------------------------------------------------------------------------------- /prepare_data/prepare_2d_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/prepare_data/prepare_2d_data.py -------------------------------------------------------------------------------- /prepare_data/reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/prepare_data/reader.py -------------------------------------------------------------------------------- /prepare_data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/prepare_data/util.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/requirements.txt -------------------------------------------------------------------------------- /save/ENetDepth-scannet20: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/save/ENetDepth-scannet20 -------------------------------------------------------------------------------- /save/ENetDepth-scannet20_summary.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/save/ENetDepth-scannet20_summary.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/train.py -------------------------------------------------------------------------------- /transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/transforms.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/montrealrobotics/ENetDepth/HEAD/utils.py --------------------------------------------------------------------------------