博主你好,我下载了你的程序以及coco的训练集、验证集进行训练。运行环境是Ubuntu-20.04,CUDA 11.2,cuDNN 8.1, tf 2.5.0,batch size设置为了5.最终在训练了5个小时之后训练结束,但在结束之后报了FileNotFoundError,很奇怪,以下是最终报错的代码:
23453/23453 [==============================] - ETA: 0s - loss: 1.6394 - rpn_class_loss_loss: 0.0734 - rpn_bbox_loss_loss: 0.8458 - mrcnn_class_loss_loss: 0.1595 - mrcnn_bbox_loss_loss: 0.2428 - mrcnn_mask_loss_loss: 0.3179Traceback (most recent call last):
File "train.py", line 286, in
model.fit(
File "/home/+++/anaconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 1201, in fit
self._eval_data_handler = data_adapter.get_data_handler(
File "/home/+++/anaconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/data_adapter.py", line 1364, in get_data_handler
return DataHandler(*args, **kwargs)
File "/home/+++/anaconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/data_adapter.py", line 1154, in init
self._adapter = adapter_cls(
File "/home/+++/anaconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/data_adapter.py", line 932, in init
super(KerasSequenceAdapter, self).init(
File "/home/+++/anaconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/data_adapter.py", line 809, in init
peek, x = self._peek_and_restore(x)
File "/home/+++/anaconda3/envs/tf2/lib/python3.8/site-packages/tensorflow/python/keras/engine/data_adapter.py", line 943, in _peek_and_restore
return x[0], x
File "/home/+++/tf2/mask-rcnn-tf2/utils/dataloader.py", line 239, in getitem
image, boxes, mask_gt, num_crowds, image_id = self.pull_item(global_index)
File "/home/+++/tf2/mask-rcnn-tf2/utils/dataloader.py", line 318, in pull_item
image = Image.open(image_path)
File "/home/+++/anaconda3/envs/tf2/lib/python3.8/site-packages/PIL/Image.py", line 3131, in open
fp = builtins.open(filename, "rb")
FileNotFoundError: [Errno 2] No such file or directory: 'datasets/coco/JPEGImages/val2017/000000558840.jpg'
2022-12-05 03:48:43.728848: W tensorflow/core/kernels/data/generator_dataset_op.cc:107] Error occurred when finalizing GeneratorDataset iterator: Failed precondition: Python interpreter state is not initialized. The process may be terminated.
[[{{node PyFunc}}]]
我查看了一下下载的coco验证集,里面确实没有000000558840.jpg这个文件,但是为什么代码中会需要这个文件呢?
谢谢博主答疑。