imagens de carregamento de cor Keras

votos
12

Eu tenho 3 pastas com imagens coloridas. Nome da pasta é o rótulo para as imagens dentro.

cls1
  |____img_0.png
  |____ ...
  |____img_n.png
cls2
  |____img_0.png
  |____ ...
  |____img_n.png   
cls3
  |____img_0.png
  |____ ...
  |____img_n.png

Eu gostaria de usar a biblioteca Keras para criar rede neural Convolucional para a classificação, mas não consigo encontrar, como criar conjunto de dados de imagens a cores. Pode me ajudar?

Publicado 07/05/2016 em 11:31
fonte usuário
Em outras línguas...                            


1 respostas

votos
12

Por favor, considere esta essência do pré-formados modelo VGG-16 com exemplo de uso o que eu acho ilustrativo:

Para resumir:

  1. Carregar suas imagens com OpenCV ou scikit-imagem, etc. (L58)
  2. Redimensionar e / ou cortar suas imagens para ajustar o tamanho de entrada (224 * 224 para VGG-16) (L58)
  3. Calcular e subtrair para cada imagem o valor médio (L59 L61 ~)
  4. Trocar a dimensão cor antes de altura e largura dimensões (L62)
  5. Se scikit-imagem que você usar, você precisa trocar os 3 canais de cor porque as imagens de carga OpenCV como BGR canais, mas scikit-imagem carregá-lo como RGB canais.
  6. Adicionar a dimensão de tamanho de lote (L63)
  7. Shuffle, partição e concatenar-los (junto dimensão 0) para formar os dados de treinamento X_train, dados de teste X_test, etc, juntamente com a verdade chão Y_train, Y_testetc.
  8. Se o seu Conjunto de dados é demasiado grande para caber na memória, use um gerador e a função fit_generatorpara fazer o treinamento em seu lugar. (Keras também tem evaluate_generatore predict_generator)
  9. Agora você está pronto para o treinamento.
Respondeu 07/07/2016 em 10:25
fonte usuário

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more