Эта проблема связана с глубоким обучением при работе с pytorch.
Решение: Проблема с форматом. Требуемый формат — double, а результат — float. Буквальный смысл такой,но формат на двойной менять не обязательно.В это время надо поставить точку останова в коде перед ошибкой и отладить.В результате я получил что изображение и img_rgb оба в тензоре .unit8, но программе нужен формат torch.float32, поэтому добавьте .float() после этих двух строк
Произойдет указанная выше ошибка, измените ее на: