来自Github的再训练车道检测模型

时间:2018-10-24 01:47:33

标签: python opencv tensorflow image-processing

我正在尝试从Github https://github.com/MaybeShewill-CV/lanenet-lane-detection重新训练该模型,该模型将在视频上覆盖道路车道线。我使用Github存储库中的模型测试了该模型:

python tools/test_lanenet.py --is_batch False --batch_size 1 
--weights_path model/tusimple_lanenet/tusimple_lanenet_vgg_2018-05-21-11-11-03.ckpt-94000 
--image_path data/tusimple_test_image/0.jpg

并且成功了。

我按照回购中的说明添加了新的训练数据,并在imagegt_image_instancegt_image_binary文件夹中添加了新图像,但出现错误。我在bash中从Github输入以下行:

python tools/train_lanenet.py --net vgg --dataset_dir data/training_data_example/

我得到的错误是:

  

cv2.error:OpenCV(3.4.2)   /用户/travis/build/skvark/opencv-python/opencv/modules/imgproc/src/resize.cpp:4044:   错误:(-215:声明失败)函数'resize'中的!ssize.empty()

有时我还会收到此错误:

  

ValueError:变量lanenet_loss / inference / encode / conv1_1 / conv / W   已经存在

我已经修改了train.txtval.txt,并更改了在计算机上本地找到的图像的文件路径。

该如何解决?

0 个答案:

没有答案