复现owod遇到的问题

Question

在复现owod时https://github.com/JosephKJ/OWOD/tree/master，对图片进行目标检测遇到了问题

/home/aiit/Downloads/OWOD-master/src/detectron2/detectron2/modeling/roi_heads/fast_rcnn.py:498: FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via `torch.serialization.add_safe_globals`. We recommend you start setting `weights_only=True` for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature.
  self.feature_store = torch.load(self.feature_store_save_loc)

gpt告诉我加上weights_only=True，可是任然报错。

Answer 1

以下回复参考：皆我百晓生、券券喵儿等免费微信小程序作答:

根据你提供的警告信息，问题在于你在使用 torch.load() 函数时未指定 weights_only 参数，因此默认使用了 False 值，这可能会导致安全隐患。当你尝试使用 weights_only=True 时仍然报错，可能是因为你的代码中存在一些与模型加载相关的其他错误或者不兼容问题。

你可以尝试以下步骤来解决这个问题：

1. 使用 weights_only=True 参数: 确保在使用 torch.load() 时设置 weights_only=True，这将仅加载模型的权重而不会加载任何其他 Python 对象或数据。修改你的代码如下：

self.feature_store = torch.load(self.feature_store_save_loc, weights_only=True)

这将确保只有模型的权重被加载，从而避免潜在的代码执行风险。

1. 检查其他可能的错误: 如果使用 weights_only=True 后仍然报错，请仔细检查其他部分的代码，特别是与模型加载和初始化相关的部分。确保所有的依赖项都已正确安装并且版本兼容。
2. 查看详细的错误信息: 如果仍然存在问题，请提供详细的错误信息，这将有助于更准确地定位问题所在。错误信息通常会提供更多关于问题的线索和上下文。
3. 更新依赖库: 如果你使用的 PyTorch 版本较旧，尝试更新到最新版本，以确保与 OWOD 项目的兼容性。有时候，库的新版本会修复一些已知的问题。

请注意，如果上述建议无法解决问题，可能需要进一步分析具体的错误信息和代码来找到解决方案。如果你能提供更多的上下文或详细的错误信息，我将能够提供更具体的帮助。