使用VGG模型训练灰度图片时报错,LeNet-5却不会
Created by: yeyupiaoling
我使用vgg训练灰度图是报错,报错如下,数据大小是没有错的,因为我改成彩色,然后乘上3,就一切正常了.
I0125 10:43:58.905412 5090 Util.cpp:166] commandline: --use_gpu=False --trainer_count=2
[CRITICAL 2018-01-25 10:43:58,913 layers.py:2714] Input layer image: Incorrect input image size 16 * 16 for input image pixels 261
Traceback (most recent call last):
File "/home/wang/ubuntu/paddlepaddle/pycharm/PycharmProjects/TestCustomData/code/train.py", line 116, in <module>
main()
File "/home/wang/ubuntu/paddlepaddle/pycharm/PycharmProjects/TestCustomData/code/train.py", line 52, in main
out = vgg.vgg16(image, class_dim=CLASS_DIM)
File "/media/wang/软件/ubuntu/paddlepaddle/pycharm/PycharmProjects/TestCustomData/code/vgg.py", line 49, in vgg16
return vgg(input, nums, class_dim)
File "/media/wang/软件/ubuntu/paddlepaddle/pycharm/PycharmProjects/TestCustomData/code/vgg.py", line 20, in vgg
conv1 = conv_block(input, 64, nums[0], 3)
File "/media/wang/软件/ubuntu/paddlepaddle/pycharm/PycharmProjects/TestCustomData/code/vgg.py", line 16, in conv_block
pool_type=paddle.pooling.Max())
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/default_decorators.py", line 53, in __wrapper__
return func(*args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/default_decorators.py", line 53, in __wrapper__
return func(*args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/networks.py", line 420, in img_conv_group
**extra_kwargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/default_decorators.py", line 53, in __wrapper__
return func(*args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/default_decorators.py", line 53, in __wrapper__
return func(*args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/default_decorators.py", line 53, in __wrapper__
return func(*args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/default_decorators.py", line 53, in __wrapper__
return func(*args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/layers.py", line 415, in wrapper
return method(*args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer_config_helpers/layers.py", line 2714, in img_conv_layer
**ExtraLayerAttribute.to_kwargs(layer_attr))
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer/config_parser.py", line 3938, in Layer
return layer_func(name, **xargs)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer/config_parser.py", line 2105, in __init__
conv_conf, num_filters)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer/config_parser.py", line 1374, in parse_conv
get_img_size(input_layer_name, conv.channels)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer/config_parser.py", line 1212, in get_img_size
% (input_layer_name, img_size, img_size_y, img_pixels))
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer/config_parser.py", line 156, in config_assert
logger.fatal(msg)
File "/usr/local/lib/python2.7/dist-packages/paddle/trainer/config_parser.py", line 4288, in my_fatal
raise Exception()
Exception