Final project codes for Computer vision in the spring semester at NYCU. This is a basic lane detection using TuSimple dataset.
- Zhi-Yi Chin [email protected]
- Shao-Yu Weng [email protected]
- Bo-Yu Cheng [email protected]
conda environment
1080Ti or 2080Ti GPU device
pytorch
torchvision
visdom
cudatoolkit=10.1
opencv
sklearn
ujson
matplotlib
2 Hourglass Network + aggregation of SGPN loss
- Method 1: Edge detection + Hough transform
- Method 2: Perspective transform