не удается визуализировать изображения во время загрузки изображений в COCO detectron2

#detectron

#детектор

Вопрос:

Я хотел загрузить и отобразить изображения с масками на detectron2. Но во время загрузки он выдает мне такую ошибку:

 6 for s in random.sample(dataset_custom, n):  7 img = cv2.imread(s['file_name']) ----gt; 8 v = Visualizer(img[:,:,::-1], metadata = dataset_custom_metadata, scale = 0.5)  9 v = v.draw_dataset_dict(s)  10 #plt.figure(figsize = (15, 20))  TypeError: 'NoneType' object is not subscriptable add Codeadd Markdown   

Кроме того, я получил такую ошибку в записных книжках colab и kaggle. В моем местном jupyter я не получил этой ошибки. Что нужно сделать, чтобы избавиться от ошибки типа? Могу ли я оставить этот шаг и не использовать визуализатор или он необходим во время сегментации?