Error in Test Mask RCNN model code

[Abhinav-974]

Im getting this error in model.py in mrcnn in Test MaskRCNN folder when I run mask rcnn predict.py -
abhinav@isropc-MS-7D98:~/Documents/MeghNA-master/Mask RCNN/Test MaskRCNN$ /bin/python3 "/home/abhinav/Documents/MeghNA-master/Mask RCNN/Test MaskRCNN/mask_rcnn_predict.py"
2024-04-21 23:02:27.771430: I tensorflow/core/util/port.cc:113] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable TF_ENABLE_ONEDNN_OPTS=0.
2024-04-21 23:02:27.796906: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 AVX_VNNI FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-04-21 23:02:28.132784: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
[INFO] loading Mask R-CNN model...
2024-04-21 23:02:28.903812: I external/local_xla/xla/stream_executor/cuda/cuda_executor.cc:998] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero. See more at https://github.com/torvalds/linux/blob/v6.0/Documentation/ABI/testing/sysfs-bus-pci#L344-L355
2024-04-21 23:02:28.936040: W tensorflow/core/common_runtime/gpu/gpu_device.cc:2251] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...
Traceback (most recent call last):
File "/home/abhinav/Documents/MeghNA-master/Mask RCNN/Test MaskRCNN/mask_rcnn_predict.py", line 33, in
model = modellib.MaskRCNN(mode="inference", config=config,model_dir=os.getcwd())
File "/home/abhinav/Documents/MeghNA-master/Mask RCNN/Test MaskRCNN/mrcnn/model.py", line 1838, in init
self.keras_model = self.build(mode=mode, config=config)
File "/home/abhinav/Documents/MeghNA-master/Mask RCNN/Test MaskRCNN/mrcnn/model.py", line 2036, in build
fpn_classifier_graph(rpn_rois, mrcnn_feature_maps, input_image_meta,
File "/home/abhinav/Documents/MeghNA-master/Mask RCNN/Test MaskRCNN/mrcnn/model.py", line 937, in fpn_classifier_graph
shared = KL.Lambda(lambda x: K.squeeze(K.squeeze(x, 3), 2),
File "/home/abhinav/.local/lib/python3.10/site-packages/keras/src/utils/traceback_utils.py", line 122, in error_handler
raise e.with_traceback(filtered_tb) from None
File "/home/abhinav/.local/lib/python3.10/site-packages/keras/src/layers/core/lambda_layer.py", line 95, in compute_output_shape
raise NotImplementedError(
NotImplementedError: Exception encountered when calling Lambda.call().

We could not automatically infer the shape of the Lambda's output. Please specify the output_shape argument for this Lambda layer.

Arguments received by Lambda.call():
• args=('<KerasTensor shape=(None, 1000, 1, 1, 1024), dtype=float32, sparse=False, name=keras_tensor_393>',)
• kwargs={'mask': 'None'}