Detectron2中运行demo.py时，报错：AssertionError: get_event_storage() has to be called inside a ‘with EventSt

布尔大学士

于 2023-05-15 15:22:28 发布

阅读量387

点赞数 1

CC 4.0 BY-SA版权

分类专栏： # Pytorch编写中遇到的问题文章标签：深度学习人工智能机器学习神经网络计算机视觉

本文链接：https://blog.csdn.net/qq_39237205/article/details/130685332

Pytorch编写中遇到的问题专栏收录该内容

63 篇文章

订阅专栏

在运行模型的demo时遇到一个错误，具体是AssertionError:get_event_storage()必须在一个withEventStorage(...)上下文中调用。为解决此问题，需要修改roi_heads.py文件，将直接调用storage=get_event_storage()的部分改为使用with语句包围，如：withEventStorage():storage=get_event_storage()。这个问题的解决方案参考自GitHub上的FSCE项目issue#37。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

训练完模型以后，运行demo/demo.py去可视化结果，结果报错：

python demo/demo.py --config-file configs/PASCAL_VOC/base-training/R101_FPN_base_training_split1.yml \--input input1.jpg


# 报错：

AssertionError: get_event_storage() has to be called inside a 'with EventStorage(...)' context!

解决方案

将
roi_ heads.py中的

storage = get_ event_storage ()

改为

with EventStorage():
	storage = get_event_storage()

参考文献：
https://github.com/megvii-research/FSCE/issues/37