如图,工程目录大致如下:
工作项目根目录:
attentive-gan-derainnet
Linux命令行直接运行:
python3 data_provider/data_feed_pipline.py --dataset_dir /home/gavin/Dataset/attentive-gan-derainnet/train_total --tfrecords_dir ./data
报错如下:
解决办法很简单:
只需要在data_feed_pipline.py加入以下代码:
import os
import sys
project = 'attentive-gan-derainnet' # 工作项目根目录
sys.path.append(os.getcwd().split(project)[0] + project)
重新运行则正常: