Team Ai
Modelpublic

hugginglearners/multi-object-classification

sourceHugging Faceupdated 4y agoView on Hugging Face
2likes
Model Card

Model description

This repo contains the trained model for Multi-object classification

Full credits go to Nhu Hoang

Motivation: Classifying multiple objects is a challenging task without using an object detection algorithm. This model was trained on resnet34 backbone and achieved a good accuracy.

Training and evaluation data

Training hyperparameters

The following hyperparameters were used during training:

HyperparametersValue
nameAdam
learning_rate3e-3
training_precisionfloat16