我有一个Detectron2 .pth模型，该模型通过位于以下位置的Detectron2工具功能成功转换为Caffe2 .pb：https://github.com/facebookresearch/detectron2/blob/master/tools/caffe2_converter.py
根据建议，在转换时使用--run-eval标志来确认结果，结果与原始detectron2结果非常相似。
要使用生成的model.pb和model_init.pb文件在新图像上进行推断，请使用以下功能： https://github.com/facebookresearch/detectron2/blob/master/detectron2/export/api.py（大部分） https://github.com/facebookresearch/detectron2/blob/master/detectron2/export/caffe2_inference.py
但是，推论结果还差得很远。有人可以建议为什么会发生这种情况吗？ Detectron2 repo说所有预处理都在caffe2脚本中完成，但是我丢失了什么吗？
我可以提供我的推断代码：
caffe2_model = Caffe2Model.load_protobuf(input_directory)
img = cv2.imread(input_image)
image = torch.as_tensor(img.astype("float32").transpose(2, 0, 1))
data = {'image': image, 'height': image.shape[1], 'width': image.shape[2]}
output = caffe2_model([data])