RuntimeError: [enforce 失败在 net.cc:69] op BBoxTransform: 为 net detectron 的输入 rpn_rois 所对应的源未知,操作器输入:"rpn_rois"输入:"bbox_pred"输入:"im_info"输出:"pred_bbox"名称:""类型:"BBoxTransform"参数 {名称:"correct_transform_coords"i:1}参数 {名称:"apply_scale"i:0}参数 {名称:"weights"浮点数:10浮点数:10浮点数:5浮点数:5}
作者: pjoshi0621创建于 2019年3月11日更新于 2023年9月22日
我正在使用 Detectron retinanet 来训练模型。经过训练，我得到一个“.pkl”文件作为输出。然后，我将其转换为“.pb”文件。并获得了两个文件“model_init.pb”和model.pb。现在，我正在尝试使用这个“.pb”文件加载模型。为此，我使用了以下代码：
内容来源: facebookresearch/Detectron