Блог

Я создаю приложение, которое сравнивает два изображения, чтобы найти сходство. Моя модель была обучена без нормализации изображения. Я хочу преобразовать изображение в тензор без нормализации. Что мне делать?

     public Tensor recognize(Bitmap image){

    //        Tensor input =  TensorImageUtils.bitmapToFloat32Tensor(
    //                image,
    //                TensorImageUtils.TORCHVISION_NORM_MEAN_RGB,
    //                TensorImageUtils.TORCHVISION_NORM_STD_RGB);

    float[] a = new float[]{1.0f, 1.0f, 1.0f};
    float[] b = new float[]{1.0f, 1.0f, 1.0f};
    Tensor input =  TensorImageUtils.bitmapToFloat32Tensor(
            image, a, b);


    Tensor output = module.forward(IValue.from(input)).toTensor();

    return output;
}