Mask_R_CNN_Keypoints

Mask_R_CNN_Keypoints copied to clipboard

Error in Softmax

10

``` ResourceExhaustedError (see above for traceback): Ran out of GPU memory when allocating 0 bytes for [[Node: mrcnn_mask_loss/SoftmaxCrossEntropyWithLogits = SoftmaxCrossEntropyWithLogits[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/gpu:0"](mrcnn_mask_loss/Reshape_3, mrcnn_mask_loss/Reshape_4)]] [[Node: training/SGD/gradients/roi_align_mask/ExpandDims_1_grad/Reshape/_1987 = _Recv[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_7770_training/SGD/gradients/roi_align_mask/ExpandDims_1_grad/Reshape",...

filipetrocadoferreira

Mask + key point

1

Hi, the authors of the original paper had masking + key point detection. Is this possible with your code?

teaglin