pytorch/examples

How to use trained model to classifier pictures?

Ouverte

#450 ouverte le 25 nov. 2018

Voir sur GitHubGuide contributeur
 (2 commentaires) (0 réaction) (0 personne assignée)Python (9 429 forks)batch import
help wantedvision

Métriques du dépôt

Stars
 (21 634 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

Description

I have trained a best model by imagenet,but code repo has given does not have test option,so how can I use the model have trained to classifier pictures with labels?

Guide contributeur

Direction de recherche
Écrivez un petit script pour charger le modèle entraîné, prétraiter les images d'entrée (par exemple redimensionner, normaliser avec les statistiques d'ImageNet), exécuter l'inférence et mapper les indices de sortie aux étiquettes de classe d'ImageNet. Utilisez torchvision.transforms de PyTorch pour le prétraitement.
Stack technique
pythonpytorch
Domaine
machine learningai
Type d'issue
Recherche
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
PythonPyTorch
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
70