kritiksoman / GIMP-ML

AI for GNU Image Manipulation Program
https://kritiksoman.github.io/GIMP-ML-Docs/index.html
MIT License
1.39k stars 126 forks source link

Errors when running deblur #72

Open mattlinux opened 3 months ago

mattlinux commented 3 months ago

I am getting the following errors. Other plugins work OK. The six module seems to be installed properly as it is reported by "pip list". Running on Windows 11

Traceback (most recent call last): File "C:\Users\marti\gimpenv3\Lib\site-packages\gimpml\plugins\deblur....\tools\deblur.py", line 35, in output = get_deblur(image, cpu_flag=force_cpu, weight_path=weight_path) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Users\marti\gimpenv3\Lib\site-packages\gimpml\plugins\deblur....\tools\deblur.py", line 18, in get_deblur predictor = Predictor( ^^^^^^^^^^ File "C:\Users\marti\gimpenv3\Lib\site-packages\gimpml\tools\DeblurGANv2\predictorClass.py", line 89, in init model = get_generator_new(weights_path[0:-11]) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Users\marti\gimpenv3\Lib\site-packages\gimpml\tools\DeblurGANv2\models\networks.py", line 420, in get_generator_new model_g = torch.load(weights_path + "mymodel.pth") ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Users\marti\gimpenv3\Lib\site-packages\torch\serialization.py", line 1040, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Users\marti\gimpenv3\Lib\site-packages\torch\serialization.py", line 1268, in _legacy_load result = unpickler.load() ^^^^^^^^^^^^^^^^ File "C:\Users\marti\gimpenv3\Lib\site-packages\torch\serialization.py", line 1073, in find_class return super().find_class(mod_name, name) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Users\marti\gimpenv3\Lib\site-packages\pretrainedmodels__init.py", line 4, in from . import datasets File "C:\Users\marti\gimpenv3\Lib\site-packages\pretrainedmodels\datasets\init__.py", line 2, in from .voc import Voc2007Classification File "C:\Users\marti\gimpenv3\Lib\site-packages\pretrainedmodels\datasets\voc.py", line 6, in from six.moves.urllib.parse import urlparse ModuleNotFoundError: No module named 'six'