Salta para o conte√ļdo

Conjunto de dados MNIST

O conjunto de dados MNIST (Modified National Institute of Standards and Technology) é uma grande base de dados de dígitos manuscritos que é normalmente utilizada para treinar vários sistemas de processamento de imagem e modelos de aprendizagem automática. Foi criado através da "re-mistura" das amostras dos conjuntos de dados originais do NIST e tornou-se uma referência para avaliar o desempenho dos algoritmos de classificação de imagens.

Características principais

  • A MNIST cont√©m 60.000 imagens de treino e 10.000 imagens de teste de d√≠gitos manuscritos.
  • O conjunto de dados inclui imagens em escala de cinzentos de tamanho 28x28 pix√©is.
  • As imagens s√£o normalizadas para caberem numa caixa delimitadora de 28x28 pix√©is e anti-aliased, introduzindo n√≠veis de escala de cinzentos.
  • A MNIST √© amplamente utilizada para treino e teste no dom√≠nio da aprendizagem autom√°tica, especialmente para tarefas de classifica√ß√£o de imagens.

Estrutura do conjunto de dados

O conjunto de dados MNIST est√° dividido em dois subconjuntos:

  1. Conjunto de treino: Este subconjunto contém 60.000 imagens de dígitos manuscritos utilizados para treinar modelos de aprendizagem automática.
  2. Conjunto de teste: Este subconjunto consiste em 10.000 imagens utilizadas para testar e avaliar os modelos treinados.

MNIST alargada (EMNIST)

O MNIST alargado (EMNIST) √© um novo conjunto de dados desenvolvido e publicado pelo NIST para suceder ao MNIST. Enquanto o MNIST inclu√≠a apenas imagens de d√≠gitos manuscritos, o EMNIST inclui todas as imagens da Base de Dados Especial 19 do NIST, que √© uma grande base de dados de letras mai√ļsculas e min√ļsculas manuscritas, bem como de d√≠gitos. As imagens da EMNIST foram convertidas para o mesmo formato de 28x28 p√≠xeis, pelo mesmo processo, que as imagens da MNIST. Por conseguinte, as ferramentas que funcionam com o conjunto de dados MNIST mais antigo e mais pequeno funcionar√£o provavelmente sem modifica√ß√Ķes com o EMNIST.

Aplica√ß√Ķes

O conjunto de dados MNIST é amplamente utilizado para treinar e avaliar modelos de aprendizagem profunda em tarefas de classificação de imagens, como Redes Neurais Convolucionais (CNNs), Máquinas de Vectores de Suporte (SVMs) e vários outros algoritmos de aprendizagem automática. O formato simples e bem estruturado do conjunto de dados torna-o um recurso essencial para investigadores e profissionais no domínio da aprendizagem automática e da visão computacional.

Utilização

Para treinar um modelo CNN no conjunto de dados MNIST para 100 épocas com um tamanho de imagem de 32x32, podes utilizar os seguintes snippets de código. Para obter uma lista abrangente dos argumentos disponíveis, consulta a página Treino do modelo.

Exemplo de comboio

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8n-cls.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="mnist", epochs=100, imgsz=32)
# Start training from a pretrained *.pt model
cnn detect train data=mnist model=yolov8n-cls.pt epochs=100 imgsz=28

Exemplos de imagens e anota√ß√Ķes

O conjunto de dados MNIST contém imagens em escala de cinzentos de dígitos manuscritos, proporcionando um conjunto de dados bem estruturado para tarefas de classificação de imagens. Aqui tens alguns exemplos de imagens do conjunto de dados:

Imagem de amostra do conjunto de dados

O exemplo mostra a variedade e complexidade dos d√≠gitos manuscritos no conjunto de dados MNIST, real√ßando a import√Ęncia de um conjunto de dados diversificado para treinar modelos robustos de classifica√ß√£o de imagens.

Cita√ß√Ķes e agradecimentos

Se utilizares o conjunto de dados MNIST no teu

trabalho de investigação ou desenvolvimento, por favor cita o seguinte artigo:

@article{lecun2010mnist,
         title={MNIST handwritten digit database},
         author={LeCun, Yann and Cortes, Corinna and Burges, CJ},
         journal={ATT Labs [Online]. Available: http://yann.lecun.com/exdb/mnist},
         volume={2},
         year={2010}
}

Gostar√≠amos de agradecer a Yann LeCun, Corinna Cortes e Christopher J.C. Burges pela cria√ß√£o e manuten√ß√£o do conjunto de dados MNIST como um recurso valioso para a comunidade de investiga√ß√£o em aprendizagem autom√°tica e vis√£o computacional. Para mais informa√ß√Ķes sobre o conjunto de dados MNIST e os seus criadores, visita o s√≠tio Web do conjunto de dados MNIST.

FAQ

O que é o conjunto de dados MNIST e porque é importante para a aprendizagem automática?

O conjunto de dados MNIST, ou conjunto de dados modificado do National Institute of Standards and Technology, √© uma cole√ß√£o de d√≠gitos manuscritos muito utilizada, concebida para treinar e testar sistemas de classifica√ß√£o de imagens. Inclui 60.000 imagens de treino e 10.000 imagens de teste, todas em escala de cinzentos e com 28x28 pixels de tamanho. A import√Ęncia do conjunto de dados reside no seu papel como refer√™ncia padr√£o para a avalia√ß√£o de algoritmos de classifica√ß√£o de imagens, ajudando investigadores e engenheiros a comparar m√©todos e a acompanhar os progressos neste dom√≠nio.

Como posso utilizar Ultralytics YOLO para treinar um modelo no conjunto de dados MNIST?

Para treinar um modelo no conjunto de dados MNIST utilizando Ultralytics YOLO , podes seguir estes passos:

Exemplo de comboio

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8n-cls.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="mnist", epochs=100, imgsz=32)
# Start training from a pretrained *.pt model
cnn detect train data=mnist model=yolov8n-cls.pt epochs=100 imgsz=28

Para obteres uma lista detalhada dos argumentos de formação disponíveis, consulta a página Formação.

Qual é a diferença entre os conjuntos de dados MNIST e EMNIST?

O conjunto de dados MNIST cont√©m apenas d√≠gitos manuscritos, enquanto o conjunto de dados MNIST alargado (EMNIST) inclui tanto d√≠gitos como letras mai√ļsculas e min√ļsculas. O EMNIST foi desenvolvido como sucessor do MNIST e utiliza o mesmo formato de 28x28 pix√©is para as imagens, tornando-o compat√≠vel com as ferramentas e modelos concebidos para o conjunto de dados MNIST original. Esta gama mais vasta de caracteres no EMNIST torna-o √ļtil para uma maior variedade de aplica√ß√Ķes de aprendizagem autom√°tica.

Posso utilizar o Ultralytics HUB para treinar modelos em conjuntos de dados personalizados como o MNIST?

Sim, podes utilizar o Ultralytics HUB para treinar modelos em conjuntos de dados personalizados como o MNIST. Ultralytics O HUB oferece uma interface fácil de usar para carregar conjuntos de dados, treinar modelos e gerenciar projetos sem a necessidade de conhecimento extenso de codificação. Para obter mais detalhes sobre como começar, consulta a página Ultralytics HUB Quickstart.



Criado em 2023-11-12, Atualizado em 2024-07-04
Autores: glenn-jocher (6)

Coment√°rios