Pycocotools eval summarize is not working in ngc pytorch image

There is a problem with cocoeval.py that the pyrokool is already installed in the pytorch image. (22.12-py3, 22.11.py3)

line 460: s = s[:,aind,mind,:,:] => s = s[:,:,:,aind,mind]

ref) cocoapi/cocoeval.py at master · cocodataset/cocoapi · GitHub

Changing the version from 2.0+nv0.2.1 to 2.0.0 worked for me.