VPGNet_Test

Forked from VPGNet

Install

由于CUDNN版本问题(7.5)，USE_CUDNN=0

make -j40
make pycaffe

1.生成数据集
目前使用caltech-lanes-dataset数据集训练，为了方便没有matlab的机器，生成lmdb数据txt已上传

cd models/vpgnet-novp/
make_lmdb.sh

PS::关于数据集，tusimple没有标注对应lane类别，BDD需要清洗
2.训练

nohup ../../build/tools/caffe train --solver=./solver.prototxt &
tail -f nohup.out

在test文件夹中提供了py和cpp实现
1.python脚本修改test.py对应文件路径
python test.py
2.cpp修改run.sh对应路径编译运行

./run.sh
./test <net.prototxt> <net.caffemodel> <inputFile_txt>

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
cmake		cmake
data		data
docs		docs
examples		examples
include/caffe		include/caffe
matlab		matlab
models		models
python		python
scripts		scripts
src		src
tools		tools
CMakeLists.txt		CMakeLists.txt
CONTRIBUTING.md		CONTRIBUTING.md
CONTRIBUTORS.md		CONTRIBUTORS.md
INSTALL.md		INSTALL.md
LICENSE		LICENSE
Makefile		Makefile
Makefile.config		Makefile.config
Makefile.config.example		Makefile.config.example
README.md		README.md
README.md.caffe		README.md.caffe
caffe.cloc		caffe.cloc