StopIteration

[yxxxxxhh111]

I encountered a problem when training maskrcnn. How can I solve it?

2023-03-13 07:59:32.738865: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX2 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2023-03-13 07:59:33.929344: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/lib/python3.9/dist-packages/cv2/../../lib64:/usr/lib64-nvidia
2023-03-13 07:59:33.929456: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/lib/python3.9/dist-packages/cv2/../../lib64:/usr/lib64-nvidia
2023-03-13 07:59:33.929476: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
read training data for maskRCNN ...
image_dir is ./content/segmentation-unet-maskrcnn/data/maskRCNN/jpg/jpg4
Traceback (most recent call last):
File "/content/segmentation-unet-maskrcnn/maskRCNN/main_train.py", line 71, in
dataset_train.load_LOL(train_dir, hill=='True')
File "/content/segmentation-unet-maskrcnn/maskRCNN/mrcnn_dataset.py", line 28, in load_LOL
image_names = next(os.walk(image_dir))[2]
StopIteration