tf.extract_image_patches для 3D изображений

Документация по tf.extract_image_patches

Это только для 2D-изображений, может ли оно быть расширено до 3D-изображений, что полезно для реализации функции потерь SSIM?

Я не могу найти исходный код. Есть похожая функция skimage.util.view_as_windowsОднако, когда я пытаюсь использовать эту функцию с тензорным потоком в качестве бэкенда в кератах, возникают ошибки. Переход от numpy array к тензорному мне сильно смутил.

0 ответов

Другие вопросы по тегам