FronkonGames/steam-games-dataset
Viewer โข Updated โข 124k โข 2.7k โข 77
How to use MAS-AI-0000/GameNet-1 with Keras:
# Available backend options are: "jax", "torch", "tensorflow".
import os
os.environ["KERAS_BACKEND"] = "jax"
import keras
model = keras.saving.load_model("hf://MAS-AI-0000/GameNet-1")
GameNet-1 is a deep learning-based computer vision system designed to recognize video games based on their cover art or in-game screenshots. Built using EfficientNet and trained on a curated dataset of popular Steam games, the model predicts both the game name and its genre(s).
EfficientNetB3 pretrained on ImageNetGlobalAveragePooling2DDropout (0.4 & 0.2)Dense(256, relu)Dense(n_classes, softmax)categorical_crossentropy lossAdam optimizer (1e-3 for frozen, 1e-5 for fine-tune)ImageDataGenerator)Base model
google/efficientnet-b3