小编Joh*_*rer的帖子

您好，尝试在 colab 中使用 Mask RCNN 模型保存的权重，但不断出现以下错误消息。我尝试了不同的方法来解决 .h5 问题，这以前是一个问题，但现在我遇到了困难。我尝试过训练模型的不同部分，但没有效果。谷歌 Colab 上没有针对这些情况提供任何具体信息。以下是引发问题的单元格：

    # Training dataset.
    dataset_train = linkedinDataset()
    dataset_train.load_dataset(dataset_dir, "train")
    dataset_train.prepare()

    # Validation dataset
    dataset_val = linkedinDataset()
    dataset_val.load_dataset(dataset_dir, "val")
    dataset_val.prepare()

    # *** This training schedule is an example. Update to your needs ***
# 
#  
#  
    print("Training network heads")
    model.train(dataset_train,
                dataset_val,
                learning_rate=config.LEARNING_RATE,
                epochs=5,
                layers='heads')```



```Training network heads
    ---------------------------------------------------------------------------
    OSError                                   Traceback (most recent call last)
    <ipython-input-19-174a93609e58> in <module>()
       17             learning_rate=config.LEARNING_RATE,
     18             epochs=5,
---> 19             layers='heads')

  2 frames
   /content/Mask_RCNN/mrcnn/model.py in train(self, train_dataset, val_dataset, learning_rate, …