Я пытаюсь переобучить эту модель с 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
, и с ней все получилось.
Я добавил новые тренировочные данные, следуя инструкции в репо, и добавил новые изображения впапки image
, gt_image_instance
и gt_image_binary
, но я получаю ошибки.Я ввожу эту строку из Github в bash:
python tools/train_lanenet.py --net vgg --dataset_dir data/training_data_example/
Я получаю следующие ошибки:
cv2.error: OpenCV (3.4.2) / Users / travis / build /skvark / opencv-python / opencv / modules / imgproc / src / resize.cpp: 4044: ошибка: (-215: утверждение не выполнено)! ssize.empty () в функции 'resize'
ииногда я также получаю эту ошибку:
ValueError: Переменная lanenet_loss / inference / encode / conv1_1 / conv / W уже существует
Я уже модифицировал train.txt
и val.txt
и изменил пути к файлам для изображений, найденных локально на моей машине.
Как это исправить?