tf-models: official.vision.detection Mask-RCNN Invalid argument: indices[1,63] = [1, -1] does not index into param shape [2,100,112,112]
问题 I am trying to train a Mask RCNN model based on the official MaskRCNN model present here: tensorflow/models. Below are the steps I followed: Created a tfrecord for training and validation. I have checked encoding and decoding of the tfrecords, it is working fine. Set up the config file as below: # my_maskrcnn.yaml train: train_file_pattern: "data/<dataset_name>/train/tfrecords/train.tfrecord-*" batch_size: 2 eval: eval_file_pattern: "data/<data_set_name>/val/tfrecords/val.tfrecord-*" batch