我已经训练了一个deeplab模型,现在在数据集上运行vis.py和eval.py文件时出现错误,
我正在使用以下脚本
python vis.py --logstostdrr --checkpoint_dir / home / ubuntu / Documents / tensorflow / models / research / deeplab / datasets / PQR / exp / train / on_trainval_set / train / --model__variant“ xception_65” --vis_crop_size 513,513- log_dir / home / ubuntu / Documents / tensorflow / models / research / deeplab / datasets / PQR / exp / train_on_trainval_set / log / --dataset_dir / home / ubuntu / Documents / tensorflow / models / research / deeplab / datasets / PQR / exp / train_on_trainval_set / test_dataset /
这里的数据集目录具有JPG图像,这是我最近创建的日志文件夹,用于在此处获取日志,并且checkpoint_dir指向训练时创建的检查点。
我已经通过传递JPEG图像,创建它们的分割图像,将它们转换为2d png文件,TF记录和检查点的方式训练了模型。
使用Tensorflow 1.15,Tesla k80 GPU,CUDA 10.0和CuDnn 7.6