gustavla / autocolorize

Automatic colorizaton of grayscale images using Deep Learning
BSD 3-Clause "New" or "Revised" License
220 stars 65 forks source link

Pip package broken: missing caffee #20

Open fengwang opened 3 years ago

fengwang commented 3 years ago

I installed version '0.2.1-00' using pip. This package depends on python python-numpy python-scikit-image python-requests python-scipy packages. But when I was trying to colorize one image, I got such an error

Traceback (most recent call last):
  File "/usr/bin/autocolorize", line 8, in <module>
    from autocolorize.__main__ import main
  File "/usr/lib/python3.9/site-packages/autocolorize/__init__.py", line 7, in <module>
    from .checker import checker_main
  File "/usr/lib/python3.9/site-packages/autocolorize/checker.py", line 5, in <module>
    import caffe
ModuleNotFoundError: No module named 'caffe'