tsurumeso / waifu2x-chainer

Chainer implementation of waifu2x
MIT License
165 stars 54 forks source link

training custom model, is this accuracy reasonable? #30

Open mugenZebra opened 4 years ago

mugenZebra commented 4 years ago

!python /content/waifu2x-chainer/train.py --gpu 0 --dataset_dir ./jpg --patches 32 --epoch 10 --model_name reference_scale_rgb --downsampling_filters box lanczos --lr_decay_interval 3 --arch UpConv7 /usr/local/lib/python3.6/dist-packages/chainer/_environment_check.py:91: UserWarning:

Multiple installations of CuPy package has been detected. You should select only one package from from ['cupy-cuda102', 'cupy-cuda101', 'cupy-cuda100', 'cupy-cuda92', 'cupy-cuda91', 'cupy-cuda90', 'cupy-cuda80', 'cupy']. Follow these steps to resolve this issue:

  1. pip list to list CuPy packages installed
  2. pip uninstall <package name> to uninstall all CuPy packages
  3. pip install <package name> to install the proper one

'''.format(name=name, pkgs=pkgs))

I read that the PSNR for the pretrained model is 30+, is 23dB too low? Maybe my training set is bad? My training set consists of mostly black and white images, specifically Japanese comic book pages, if that matters.