Kaludi/data-csgo-weapon-classification
Viewer β’ Updated β’ 1.65k β’ 103 β’ 1
How to use Kaludi/csgo-weapon-classification with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-classification", model="Kaludi/csgo-weapon-classification")
pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png") # Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("Kaludi/csgo-weapon-classification")
model = AutoModelForImageClassification.from_pretrained("Kaludi/csgo-weapon-classification")This is a CSGO Weapon Classifier Model that has been trained by Kaludi to recognize 11 different types of Counter-Strike: Global Offensive (CSGO) Weapons, which include AK-47,AWP,Famas,Galil-AR,Glock,M4A1,M4A4,P-90,SG-553,UMP,USP. The model is capable of accurately classifying the weapon name present in an image. With its deep understanding of the characteristics of each weapon in the game, the model is a valuable tool for players and fans of CSGO.
Tis model supports a Gradio Web UI to run the csgo-weapon-classification model: