sharktide/FireTrustNet

This model requires the following custom objects:

import tensorflow as tf
from tensorflow.keras.saving import register_keras_serializable
from tensorflow.keras import layers, models, backend as K
import numpy as np

@register_keras_serializable()
def firetrust_activation(x):
    return 0.5 + tf.sigmoid(x)  # output in range [0.5, 1.5]