MaybeShewill-CV / attentive-gan-derainnet

Unofficial tensorflow implemention of "Attentive Generative Adversarial Network for Raindrop Removal from A Single Image (CVPR 2018) " model https://maybeshewill-cv.github.io/attentive-gan-derainnet/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

no pretrained model "vgg16.npy"

JerryChang305 opened this issue · comments

HI

When I run train_model.py, I got an error

Traceback (most recent call last):
File "train_model.py", line 329, in
train_model(args.dataset_dir, weights_path=args.weights_path)
File "train_model.py", line 264, in train_model
pretrained_weights = np.load('./data/vgg16.npy', encoding='latin1').item()
File "/home/jerry/anaconda3/envs/attention_map/lib/python3.6/site-packages/numpy/lib/npyio.py", line 384, in load
fid = open(file, "rb")
FileNotFoundError: [Errno 2] No such file or directory: './data/vgg16.npy'

there is no data/vgg16.npy file in the data folder
where should I find it

thanks for the great work!