slavija-b / depth_cnn Goto Github PK
View Code? Open in Web Editor NEWThis project forked from erithwik/depth-detection-cnn
A convolutional neural network that takes in two stereo images and outputs a depth map with rather reasonable accuracy. The dataset is taken from https://github.com/LouisFoucard/DepthMap_dataset, and the model (15_epoch_model.h5) is only trained through 15 epochs but still produces a believable outcome.