milan@ubuntu:~/Desktop/interactive-deep-colorization-master/interactive-deep-colorization-master$ python ideepcolor.py --0 [GPU_ID]
/home/milan/Desktop/interactive-deep-colorization-master/interactive-deep-colorization-master/ui/gui_draw.py:22: UserWarning:
This call to matplotlib.use() has no effect because the backend has already
been chosen; matplotlib.use() must be called *before* pylab, matplotlib.pyplot,
or matplotlib.backends is imported for the first time.
The backend was *originally* set to 'TkAgg' by the following code:
File "ideepcolor.py", line 8, in <module>
from ui import gui_design
File "/home/milan/Desktop/interactive-deep-colorization-master/interactive-deep-colorization-master/ui/gui_design.py", line 3, in <module>
from . import gui_draw
File "/home/milan/Desktop/interactive-deep-colorization-master/interactive-deep-colorization-master/ui/gui_draw.py", line 19, in <module>
import matplotlib.pyplot as plt
File "/home/milan/.local/lib/python2.7/site-packages/matplotlib/pyplot.py", line 71, in <module>
from matplotlib.backends import pylab_setup
File "/home/milan/.local/lib/python2.7/site-packages/matplotlib/backends/__init__.py", line 16, in <module>
line for line in traceback.format_stack()
mpl.use('Qt4Agg')
Traceback (most recent call last):
File "ideepcolor.py", line 9, in <module>
from data import colorize_image as CI
File "/home/milan/Desktop/interactive-deep-colorization-master/interactive-deep-colorization-master/data/colorize_image.py", line 6, in <module>
import caffe
uradim sve korake u terminalu i prijavi mi
kontam da me ovo zeza:
Run the UI: python ideepcolor.py --gpu [GPU_ID]. Arguments are described below:
--win_size [512] GUI window size
--gpu [0] GPU number
--image_file ['./test_imgs/mortar_pestle.jpg'] path to the image file
ok, kad pokušam preko cpua da pokrenem, prateći
In case it helps someone here is what i did to get it running on debian stretch with python 3 (debians repository only has the caffe python module for python3).
If you want to use the notebooks you need some other stuff, but ideepcolor.py should work with this.
Install caffe
Make sure that your /etc/apt/sources.list contains contrib and non-free sections if you want to install the CUDA version, for instance:
deb
http://ftp2.cn.debian.org/debian sid main contrib non-free
Then we update APT cache and directly install Caffe. Note, the cpu version and the cuda version cannot coexist.
sudo apt update
sudo apt install [ caffe-cpu | caffe-cuda ]
Install PyQt and sklearn
sudo apt-get install python3-pyqt4 python3-sklearn python3-skimage
In case you don't have git, pip3 or wget already installed:
sudo apt-get install git python3-pip wget
Install qdarkstyle package
sudo pip3 install qdarkstyle
Install Opencv
sudo pip3 install opencv-python
Download and start ideepcolor
mkdir ideepcolor
cd ideepcolor
# Use a patched version compatible with python3
git clone -b python3
https://github.com/SleepProgger/interactive-deep-colorization
cd interactive-deep-colorization/
# Fetch models
sh models/fetch_models.sh
Start ideepcolor.py for CPU usage
python3 ideepcolor.py --cpu_mode
milan@ubuntu:~/Desktop/interactive-deep-colorization-master/interactive-deep-colorization-master$ python3 ideepcolor.py --cpu_mode
/usr/lib/python3/dist-packages/matplotlib/__init__.py:1403: UserWarning: This call to matplotlib.use() has no effect
because the backend has already been chosen;
matplotlib.use() must be called *before* pylab, matplotlib.pyplot,
or matplotlib.backends is imported for the first time.
warnings.warn(_use_error_msg)
[win_size] = 512
[image_file] = test_imgs/mortar_pestle.jpg
[gpu] = 0
[cpu_mode] = True
[color_prototxt] = ./models/reference_model/deploy_nodist.prototxt
[color_caffemodel] = ./models/reference_model/model.caffemodel
[dist_prototxt] = ./models/reference_model/deploy_nopred.prototxt
[dist_caffemodel] = ./models/reference_model/model.caffemodel
[load_size] = 256
ColorizeImageCaffe instantiated
gpu_id = -1, net_path = ./models/reference_model/deploy_nodist.prototxt, model_path = ./models/reference_model/model.caffemodel
WARNING: Logging before InitGoogleLogging() is written to STDERR
W0422 01:02:05.291738 15655 _caffe.cpp:139] DEPRECATION WARNING - deprecated use of Python interface
W0422 01:02:05.291760 15655 _caffe.cpp:140] Use this instead (with the named "weights" parameter):
W0422 01:02:05.291764 15655 _caffe.cpp:142] Net('./models/reference_model/deploy_nodist.prototxt', 1, weights='./models/reference_model/model.caffemodel')
Traceback (most recent call last):
File "ideepcolor.py", line 53, in <module>
colorModel.prep_net(args.gpu, args.color_prototxt, args.color_caffemodel)
File "/home/milan/Desktop/interactive-deep-colorization-master/interactive-deep-colorization-master/data/colorize_image.py", line 235, in prep_net
self.net = caffe.Net(prototxt_path, caffemodel_path, caffe.TEST)
RuntimeError: Could not open file ./models/reference_model/model.caffemodel
svaka pomoć dobrodošla!
) za virtuelku
