MatplotlibDeprecationWarning:
The support for Qt4 was deprecated in Matplotlib 3.3 and will be removed two minor releases later.
from matplotlib.backends.qt_compat import QtGui
[win_size] = 512
[image_file] = test_imgs/mortar_pestle.jpg
[gpu] = 0
[cpu_mode] = False
[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
[color_model] = ./models/pytorch/caffemodel.pth
[backend] = pytorch
[pytorch_maskcent] = False
[load_size] = 256
ColorizeImageTorch instantiated
path = ./models/pytorch/caffemodel.pth
Model set! dist mode? False
ColorizeImageTorch instantiated
path = ./models/pytorch/caffemodel.pth
Model set! dist mode? True
test_imgs/mortar_pestle.jpg
scale = 2.000000
But the GPU does nothing, while the CPU is maxed out and the colorization is very slow.
I have installed all of the GPU stuff and verified it's working, so I don't see what the problem is.
Here's the output it gives when run:
MatplotlibDeprecationWarning: The support for Qt4 was deprecated in Matplotlib 3.3 and will be removed two minor releases later. from matplotlib.backends.qt_compat import QtGui [win_size] = 512 [image_file] = test_imgs/mortar_pestle.jpg [gpu] = 0 [cpu_mode] = False [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 [color_model] = ./models/pytorch/caffemodel.pth [backend] = pytorch [pytorch_maskcent] = False [load_size] = 256 ColorizeImageTorch instantiated path = ./models/pytorch/caffemodel.pth Model set! dist mode? False ColorizeImageTorch instantiated path = ./models/pytorch/caffemodel.pth Model set! dist mode? True test_imgs/mortar_pestle.jpg scale = 2.000000
But the GPU does nothing, while the CPU is maxed out and the colorization is very slow. I have installed all of the GPU stuff and verified it's working, so I don't see what the problem is.