FronkonGames/steam-games-dataset
Viewer โข Updated โข 124k โข 2.15k โข 91
How to use MAS-AI-0000/GameNet-1 with Keras:
# !pip install -U keras tensorflow huggingface_hub
# Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here;
# "jax" and "torch" also work for computation once TensorFlow is installed.
import os
os.environ["KERAS_BACKEND"] = "tensorflow"
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
# !pip install -U keras tensorflow huggingface_hub # Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here; # "jax" and "torch" also work for computation once TensorFlow is installed. import os os.environ["KERAS_BACKEND"] = "tensorflow" import keras model = keras.saving.load_model("hf://MAS-AI-0000/GameNet-1")