Views
No views yet
mobilenetv3_small_screenshot.keras — trained Keras model for screenshot / not_screenshot classification1import tensorflow as tf
2from tensorflow import keras
3
4model = keras.models.load_model("mobilenetv3_small_screenshot.keras")
5
6# Example inference
7img = keras.utils.load_img("image.png", target_size=(224, 224))
8arr = keras.utils.img_to_array(img)
9arr = tf.keras.applications.mobilenet_v3.preprocess_input(arr)
10arr = arr[None, ...]
11pred = model.predict(arr)
12score = float(pred[0, 0])
13label = "screenshot" if score >= 0.5 else "not_screenshot"
14print(label, score)224x224score >= 0.5 → screenshotscore < 0.5 → not_screenshotMobileNetV3Small pretrained on ImageNetsigmoidbinary_crossentropy