Final project for EE 511 Statistical Learning
By Sandesh Banskota and Chau Huynh
This is a CNN algorithm to determine drivable areas by labeling pixels on imagesfrom the Berkeley Deep Drive dataset https://bdd-data.berkeley.edu/ (github repo: https://github.com/ucbdrive/bdd100k/tree/master/bdd100k)
Input: bdd100k/images Label: bdd100k/drivable_maps/color_labels
References:
U-Net: https://arxiv.org/pdf/1505.04597.pdf Github repo: https://github.com/milesial/Pytorch-UNet/tree/master/unet
FCN: https://arxiv.org/pdf/1411.4038.pdf Github repo: https://github.com/wkentaro/pytorch-fcn/tree/master/torchfcn/models