python - 我想在不更改源文件的情况下更改 python 中的导入函数
问题描述
我想在 python 中更改一个导入的函数,但实际上不知道从哪里开始或如何做到这一点。我以前见过装饰器和钩子,但不确定这是否是解决问题的正确方法
基本上我要更改的功能如下:
def load_coco_json(json_file, image_root, dataset_name=None, extra_annotation_keys=None):
这个函数做了很多事情,但我感兴趣的是
img_ids = sorted(coco_api.imgs.keys())
我想添加某种装饰器来完成以下任务:
img_ids = sorted(coco_api.imgs.keys())
img_ids = img_ids[0:20]
基本上,从我的处理笔记本中,我希望能够定义在这个 load_coco_json 函数中使用了多少 img_id。任何帮助,将不胜感激!
解决方案
我假设您将无法从函数中提取中间值。我所能建议的就是编写你自己的函数来返回你需要的东西。这是一个例子,它可能需要改进:
from pycocotools.coco import COCO
import contextlib
from detectron2.utils.file_io import PathManager
import io
def my_func(json_file):
json_file = PathManager.get_local_path(json_file)
with contextlib.redirect_stdout(io.StringIO()):
coco_api = COCO(json_file)
img_ids = sorted(coco_api.imgs.keys())
return img_ids