Yeah... I got this answer too. You need to build a custom loader - this should help.
import tensorflow as tf
from keras.layers import DepthwiseConv2D
from keras.models import load_model
import cv2
import numpy as np
# Define a custom DepthwiseConv2D class without the groups parameter
class CustomDepthwiseConv2D(DepthwiseConv2D):
def __init__(self, **kwargs):
# Remove the 'groups' parameter if it exists
if 'groups' in kwargs:
del kwargs['groups'] # Remove the groups parameter
super().__init__(**kwargs)
# Create a dictionary of custom objects to pass to the load_model function
custom_objects = {
'DepthwiseConv2D': CustomDepthwiseConv2D,
}
# Load the model with the custom object
try:
model = load_model("keras_model.h5", custom_objects=custom_objects,
compile=False)
print("Model loaded successfully.")
except Exception as e:
print(f"Error loading model: {e}")