使用detectron2转Mask R-CNN模型为pb及Colab数据集注册报错求助

问题1：使用model_convert.py将pth格式Mask R-CNN模型转pb失败 #

• 先确认转换用的detectron2版本和训练模型时的版本完全一致，跨版本API变动会导致模型结构解析不匹配，直接触发报错。
• 转换前先写简易测试脚本加载pth权重完成单图推理，确认权重文件本身没有损坏、配置参数和训练时完全对齐，再执行转换操作。
• 检查转换命令参数，--config-file指定的配置文件必须和训练用配置完全一致，自定义参数比如MODEL.ROI_HEADS.NUM_CLASSES不能出现偏差。
• 如果报错提示存在不支持的算子，可先通过官方导出脚本将pth转为onnx中间格式，再从onnx转pb，规避直接转换的兼容性问题。

问题2：Colab环境已执行register_coco_instances仍提示数据集未注册 #

• 确认register_coco_instances的代码块执行顺序早于加载数据集的代码块，Colab分块运行时如果先执行加载逻辑、再执行注册逻辑，或者注册代码块运行报错未完成执行，都会触发该错误。
• 核对注册时传入的数据集名称和调用时的名称拼写完全一致，detectron2数据集名称大小写敏感，下划线、大小写的差异都会导致匹配失败。
• 注册代码执行后新增一行print(DatasetCatalog.list())，查看输出列表中是否存在fruits_nuts，如果不存在说明注册逻辑本身未生效，重点检查传入的json标注路径、图片存储路径在Colab环境中是否真实存在，路径错误会导致注册静默失败。
• Colab运行时重启后所有已注册的数据集都会被清空，必须重新执行注册代码才能正常调用。

内容的提问来源于stack exchange，提问作者Sobel Li