├── .gitignore ├── LICENSE ├── README.md ├── dataloaders ├── dataloader.py ├── nyu.py └── transforms.py ├── deploy ├── data │ ├── depth.npy │ ├── depth.png │ ├── pred.npy │ ├── pred.png │ ├── rgb.npy │ ├── rgb.png │ └── visualize.py └── tx2_run_tvm.py ├── imagenet ├── __init__.py └── mobilenet.py ├── img ├── acc_fps_cpu.png ├── acc_fps_gpu.png └── visualization.png ├── main.py ├── metrics.py ├── models.py ├── tvm_compile └── tuning │ ├── tx2-cpu.mobilenet-nnconv5.trials=1000.stop=250.log │ ├── tx2-cpu.mobilenet-nnconv5dw-skipadd-pruned.trials=1000.stop=250.log │ ├── tx2-cpu.mobilenet-nnconv5dw-skipadd.trials=1000.stop=250.log │ ├── tx2-cpu.mobilenet-nnconv5dw.trials=1000.stop=250.log │ ├── tx2-gpu.mobilenet-nnconv5.trials=2000.stop=600.log │ ├── tx2-gpu.mobilenet-nnconv5dw-skipadd-pruned.trials=2000.stop=600.log │ ├── tx2-gpu.mobilenet-nnconv5dw-skipadd.trials=2000.stop=600.log │ └── tx2-gpu.mobilenet-nnconv5dw.trials=2000.stop=600.log └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | *.pyc 3 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/README.md -------------------------------------------------------------------------------- /dataloaders/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/dataloaders/dataloader.py -------------------------------------------------------------------------------- /dataloaders/nyu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/dataloaders/nyu.py -------------------------------------------------------------------------------- /dataloaders/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/dataloaders/transforms.py -------------------------------------------------------------------------------- /deploy/data/depth.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/deploy/data/depth.npy -------------------------------------------------------------------------------- /deploy/data/depth.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/deploy/data/depth.png -------------------------------------------------------------------------------- /deploy/data/pred.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/deploy/data/pred.npy -------------------------------------------------------------------------------- /deploy/data/pred.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/deploy/data/pred.png -------------------------------------------------------------------------------- /deploy/data/rgb.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/deploy/data/rgb.npy -------------------------------------------------------------------------------- /deploy/data/rgb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/deploy/data/rgb.png -------------------------------------------------------------------------------- /deploy/data/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/deploy/data/visualize.py -------------------------------------------------------------------------------- /deploy/tx2_run_tvm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/deploy/tx2_run_tvm.py -------------------------------------------------------------------------------- /imagenet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imagenet/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/imagenet/mobilenet.py -------------------------------------------------------------------------------- /img/acc_fps_cpu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/img/acc_fps_cpu.png -------------------------------------------------------------------------------- /img/acc_fps_gpu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/img/acc_fps_gpu.png -------------------------------------------------------------------------------- /img/visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/img/visualization.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/main.py -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/metrics.py -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/models.py -------------------------------------------------------------------------------- /tvm_compile/tuning/tx2-cpu.mobilenet-nnconv5.trials=1000.stop=250.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/tvm_compile/tuning/tx2-cpu.mobilenet-nnconv5.trials=1000.stop=250.log -------------------------------------------------------------------------------- /tvm_compile/tuning/tx2-cpu.mobilenet-nnconv5dw-skipadd-pruned.trials=1000.stop=250.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/tvm_compile/tuning/tx2-cpu.mobilenet-nnconv5dw-skipadd-pruned.trials=1000.stop=250.log -------------------------------------------------------------------------------- /tvm_compile/tuning/tx2-cpu.mobilenet-nnconv5dw-skipadd.trials=1000.stop=250.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/tvm_compile/tuning/tx2-cpu.mobilenet-nnconv5dw-skipadd.trials=1000.stop=250.log -------------------------------------------------------------------------------- /tvm_compile/tuning/tx2-cpu.mobilenet-nnconv5dw.trials=1000.stop=250.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/tvm_compile/tuning/tx2-cpu.mobilenet-nnconv5dw.trials=1000.stop=250.log -------------------------------------------------------------------------------- /tvm_compile/tuning/tx2-gpu.mobilenet-nnconv5.trials=2000.stop=600.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/tvm_compile/tuning/tx2-gpu.mobilenet-nnconv5.trials=2000.stop=600.log -------------------------------------------------------------------------------- /tvm_compile/tuning/tx2-gpu.mobilenet-nnconv5dw-skipadd-pruned.trials=2000.stop=600.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/tvm_compile/tuning/tx2-gpu.mobilenet-nnconv5dw-skipadd-pruned.trials=2000.stop=600.log -------------------------------------------------------------------------------- /tvm_compile/tuning/tx2-gpu.mobilenet-nnconv5dw-skipadd.trials=2000.stop=600.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/tvm_compile/tuning/tx2-gpu.mobilenet-nnconv5dw-skipadd.trials=2000.stop=600.log -------------------------------------------------------------------------------- /tvm_compile/tuning/tx2-gpu.mobilenet-nnconv5dw.trials=2000.stop=600.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/tvm_compile/tuning/tx2-gpu.mobilenet-nnconv5dw.trials=2000.stop=600.log -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dwofk/fast-depth/HEAD/utils.py --------------------------------------------------------------------------------