Solución alternativa funcional API Keras para predict_classes ()

Consulteaquí para mi pregunta anterior para obtener información de fondo. Segúnresponder sugerido porNassim Ben. Entrené el modelo de arquitectura de dos caminos usando API funcional. Ahora me siento atrapado ya que necesito predecir la clase de cada píxel. Aquí está el código para el mismo:

    imgs = io.imread(test_img).astype('float').reshape(5,240,240)
    plist = []

 # create patches from an entire slice
            for img in imgs[:-1]:
                if np.max(img) != 0:
                    img /= np.max(img)
                p = extract_patches_2d(img, (33,33))
                plist.append(p)
            patches = np.array(zip(np.array(plist[0]), np.array(plist[1]), np.array(plist[2]), np.array(plist[3])))

    # predict classes of each pixel based on model
            full_pred = self.model_comp.predict_classes(patches)
            fp1 = full_pred.reshape(208,208)

Pero según elgithub-link predic_classes () no está disponible. Entonces, mi pregunta, ¿hay alguna otra alternativa que pueda probar?

Respuestas a la pregunta(2)

Su respuesta a la pregunta