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

關於訓練數據的問題(About train dataset size problem)

Shaiken opened this issue · comments

您好,
若是訓練數據大小的問題,我該怎麼調整,除了去config調整訓練影像高寬大小和crop影像高寬大小

直接改會出現以下相關問題

hello, if training dataset size problem, how edit config .

Original stack trace for 'Reshape':
  File "tools/train_model.py", line 332, in <module>
    train_model(args.dataset_dir, weights_path=args.weights_path)
  File "tools/train_model.py", line 171, in train_model
    train_label_tensor_scale, train_net_output_tensor_scale, max_val=255),
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/ops/image_ops_impl.py", line 3370, in ssim
...

要在產dataset前改大小