2016-11-23 1 views
0

3d CNN을 실현하기 위해 cifar10 예제를 변경하려고했습니다. 그러나 회귀와 풀을 제외하고 나는 바꿔야 할 다른 것들이있는 것 같습니다. 컨볼루션 레이어, 풀 레이어 및 입력을 변경했을 때 다음 오류가 발생했습니다. 내가해야 할 일에 대한 제안이나 예가 있습니까? 감사합니다. .tensorflow에서 3d CNN의 예. InvalidArgumentError (역 추적에 대해서는 위 참조) : Tensor는 마지막 희미하게 1, 3 또는 4로 4-D가되어야합니다.

File "/mnt/disk1/Fu/Function/Tensorflow/Test/cifar10.py", line 152, in distorted_inputs 
    batch_size=FLAGS.batch_size) 
    File "/mnt/disk1/Fu/Function/Tensorflow/Test/cifar10_input.py", line 125, in distorted_inputs 
    shuffle = False) 
    File "/mnt/disk1/Fu/Function/Tensorflow/Test/cifar10_input.py", line 88, in _generate_image_and_label_batch 
    tf.image_summary('images', images) 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/logging_ops.py", line 151, in image_summary 
    tag=tag, tensor=tensor, max_images=max_images, name=scope) 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/gen_logging_ops.py", line 161, in _image_summary 
    name=name) 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/op_def_library.py", line 749, in apply_op 
    op_def=op_def) 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 2380, in create_op 
    original_op=self._default_original_op, op_def=op_def) 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 1298, in __init__ 
    self._traceback = _extract_stack() 

InvalidArgumentError (see above for traceback): Tensor must be 4-D with last dim 1, 3, or 4, not [20,32,32,32,1] 
     [[Node: ImageSummary = ImageSummary[T=DT_FLOAT, bad_color=Tensor<type: uint8 shape: [4] values: 255 0 0...>, max_images=3, _device="/job:localhost/replica:0/task:0/cpu:0"](ImageSummary/tag, batch)]] 
     [[Node: local4/weight_loss/_104 = _Send[T=DT_FLOAT, client_terminated=false, recv_device="/job:localhost/replica:0/task:0/cpu:0", send_device="/job:localhost/replica:0/task:0/gpu:0", send_device_incarnation=1, tensor_name="edge_48_local4/weight_loss", _device="/job:localhost/replica:0/task:0/gpu:0"](local4/weight_loss)]] 

답변

0

이미지 요약을 변경하여 3D 이미지를 지원하거나 그래프에서 쉽게 제거 할 수 있습니다.

관련 문제