Problems encountered in visualization

[alphacyp]

After running the model, I want to use the “demo/demo.py” to randomly input a picture for detection.

python demo/demo.py --config-file configs/PASCAL_VOC/base-training/R101_FPN_base_training_split1.yml \--input input1.jpg
Here I can normally detect the results and output a picture with the detection results.But，in the fine tune model，after I do this
python demo/demo.py --config-file configs/PASCAL_VOC/split1/10shot_CL_IoU.yml \--input input1.jpg
The content of the error report is
AssertionError: get_event_storage() has to be called inside a 'with EventStorage(...)' context!
I appreciate anyone who can help me.

[alphacyp]

Or, who has tested the test results with a random picture after running all the models?Thanks！

[NeuroDong]

Put storage = get_ event_storage () in roi_ heads.py Change to:
with EventStorage():
storage = get_event_storage()

[alphacyp]

Put storage = get_ event_storage () in roi_ heads.py Change to:
with EventStorage():
storage = get_event_storage()

Thank you very much！

[zzzzjx]

Put storage = get_ event_storage () in roi_ heads.py Change to:
with EventStorage():
storage = get_event_storage()

Thank you very much！

I have tried this but occur another errpr, how to define EventStorage?