File size: 360 Bytes
27eede3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import os
import numpy as np
import cv2
import tensorflow as tf
import streamlit as st
from tensorflow.keras.models import load_model
model = "AROI_image_segmentation.keras"
# Load the model
model = load_model(model)
print(f"The model loaded successfully")
model.compile(
loss = 'categorical_crossentropy',
optimizer = tf.keras.optimizers.Adam(lr))
|