运行maskrcnn 的demo.py时候发生的问题
Caused by op 'res2a_branch2a/convolution', defined at:
File "E:/MyCode/Python/Mask_RCNN2/samples/demo.py", line 47, in
model = modellib.MaskRCNN(mode="inference", model_dir=MODEL_DIR, config=config)
File "E:\MyCode\Python\Mask_RCNN2\mrcnn\model.py", line 1837, in init
self.keras_model = self.build(mode=mode, config=config)
File "E:\MyCode\Python\Mask_RCNN2\mrcnn\model.py", line 1901, in build
stage5=True, train_bn=config.TRAIN_BN)
File "E:\MyCode\Python\Mask_RCNN2\mrcnn\model.py", line 185, in resnet_graph
x = conv_block(x, 3, [64, 64, 256], stage=2, block='a', strides=(1, 1), train_bn=train_bn)
File "E:\MyCode\Python\Mask_RCNN2\mrcnn\model.py", line 149, in conv_block
name=conv_name_base + '2a', use_bias=use_bias)(input_tensor)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\keras\engine\base_layer.py", line 451, in call
output = self.call(inputs, **kwargs)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\keras\layers\convolutional.py", line 171, in call
dilation_rate=self.dilation_rate)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\keras\backend\tensorflow_backend.py", line 3940, in conv2d
data_format=tf_data_format)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\tensorflow\python\ops\nn_ops.py", line 754, in convolution
return op(input, filter)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\tensorflow\python\ops\nn_ops.py", line 838, in call
return self.conv_op(inp, filter)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\tensorflow\python\ops\nn_ops.py", line 502, in call
return self.call(inp, filter)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\tensorflow\python\ops\nn_ops.py", line 190, in call
name=self.name)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\tensorflow\python\ops\gen_nn_ops.py", line 725, in conv2d
data_format=data_format, dilations=dilations, name=name)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\tensorflow\python\framework\ops.py", line 3160, in create_op
op_def=op_def)
File "D:\Anaconda\envs\MaskRCNN\lib\site-packages\tensorflow\python\framework\ops.py", line 1625, in init
self._traceback = self._graph._extract_stack() # pylint: disable=protected-access
InternalError (see above for traceback): Blas SGEMM launch failed : m=65536, n=64, k=64
[[Node: res2a_branch2a/convolution = Conv2D[T=DT_FLOAT, data_format="NHWC", dilations=[1, 1, 1, 1], padding="VALID", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](max_pooling2d_1/MaxPool, res2a_branch2a/kernel/read)]]
[[Node: mrcnn_detection/ArgMax/_4221 = _Recv]]