/FCN_KITTI

Implementation of semantic segmentation of FCN structure using KITTI road dataset😝😝😝

Primary LanguagePythonMIT LicenseMIT

中文

Using FCN for road segmentation (KITTI dataset)

blob

Install

git clone https://github.com/Phoenix8215/FCN_KITTI  # clone
cd 
pip install -r requirements.txt  # install

Instructure manual

  • usemake_tsv.pyto generate the train set,validation set and test set with tsv format，the set of validation and testing is 80%,20% respectively.

  • backbone use VGG16，loss function is binary cross entropy，training 60 epochs.

  • usepython3 train.pyto train the model，and visual the process of training.

blob

  • use python3 test.pyto predict ,outputs are stored inoutputfolder.

  • generated weight files which is inckptfolder.