pytorch/examples

How to use trained model to classifier pictures?

Open

#450 aperta il 25 nov 2018

Vedi su GitHub
 (2 commenti) (0 reazioni) (0 assegnatari)Python (9429 fork)batch import
help wantedvision

Metriche repository

Star
 (21.634 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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?

Guida contributor

Direzione di ricerca
Scrivi un piccolo script per caricare il modello addestrato, preprocessare le immagini di input (ad esempio ridimensionare, normalizzare con le statistiche di ImageNet), eseguire l'inferenza e mappare gli indici di output alle etichette delle classi di ImageNet. Usa torchvision.transforms di PyTorch per il preprocessing.
Tech stack
pythonpytorch
Dominio
machine learningai
Tipo issue
Ricerca
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonPyTorch
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
70