richzhang / colorization

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.
http://richzhang.github.io/colorization/
BSD 2-Clause "Simplified" License
3.34k stars 923 forks source link

Training crash #16

Closed haihaoshen closed 7 years ago

haihaoshen commented 7 years ago

Hi,

I used my own data to train the net, but got the below error message. Could you please show some hints? Thanks.

I1203 01:20:34.406128 11072 net.cpp:165] Memory required for data: 7864320 I1203 01:20:34.406141 11072 layer_factory.hpp:77] Creating layer nongray_mask Traceback (most recent call last): File "/tmp/colorization/resources/caffe_traininglayers.py", line 4, in import numpy as np File "/usr/local/lib/python2.7/dist-packages/numpy/init.py", line 142, in from . import add_newdocs File "/usr/local/lib/python2.7/dist-packages/numpy/add_newdocs.py", line 13, in from numpy.lib import add_newdoc File "/usr/local/lib/python2.7/dist-packages/numpy/lib/init.py", line 8, in from .type_check import File "/usr/local/lib/python2.7/dist-packages/numpy/lib/type_check.py", line 11, in import numpy.core.numeric as _nx File "/usr/local/lib/python2.7/dist-packages/numpy/core/init.py", line 58, in from numpy.testing.nosetester import _numpy_tester File "/usr/local/lib/python2.7/dist-packages/numpy/testing/init.py", line 12, in from . import decorators as dec File "/usr/local/lib/python2.7/dist-packages/numpy/testing/decorators.py", line 21, in from .utils import SkipTest File "/usr/local/lib/python2.7/dist-packages/numpy/testing/utils.py", line 15, in from tempfile import mkdtemp, mkstemp File "/usr/lib/python2.7/tempfile.py", line 32, in import io as _io File "/tmp/colorization/caffe-colorization/python/caffe/io.py", line 2, in import skimage.io File "/usr/lib/python2.7/dist-packages/skimage/init.py", line 166, in from .util.dtype import File "/usr/lib/python2.7/dist-packages/skimage/util/init.py", line 1, in from .dtype import (img_as_float, img_as_int, img_as_uint, img_as_ubyte, File "/usr/lib/python2.7/dist-packages/skimage/util/dtype.py", line 8, in dtyperange = {np.bool: (False, True), AttributeError: 'module' object has no attribute 'bool_'

haihaoshen commented 7 years ago

after set correct PYTHONPATH, the issue solved.