Я закончил с сегментацией изображения на изображении / видео, теперь я не нашел никакой подсказки, чтобы удалить фон этого сегментированного изображения / видео.
В материальный репо они используйте следующий код для удаления сегментированного изображения фона
def segment(image, r):
idx = r['scores'].argmax()
mask = r['masks'][:,:,idx]
mask = np.stack((mask,)*3, axis=-1)
mask = mask.astype('uint8')
bg = 255 - mask * 255
mask_img = image*mask
result = mask_img+ bg
return result
segmentation = segment(image, r)
plt.subplots(1, figsize=(16, 16))
plt.axis('off')
plt.imshow(np.concatenate([image, segmentation], axis = 1))
можно ли использовать то же самое для видео для удаления фона?