Skip to content

Repositório utilizado para manter o projeto das discplinhas ECOM178 - REDES NEURAIS E APRENDIZADO PROFUNDO e ECOM181 - FUNDAMENTOS DE IA APLICADOS AO DIAGNÓSTICO MÉDICO.

License

Notifications You must be signed in to change notification settings

mrr00b00t/efn-siim-covid19

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

EfficientNet V1 fine-tuning para classificação de radiografias COVID19

License: MIT

Repositório para o projeto das disciplinas ECOM178-M REDES NEURAIS E APRENDIZADO PROFUNDO e ECOM181 - FUNDAMENTOS DE IA APLICADOS AO DIAGNÓSTICO MÉDICO.

Neste trabalho, foi realizado um fine-tuning da rede neural EfficientNetB5, com os pesos treinados no dataset ImageNet. Para o problema deste trabalho, o fine-tuning foi feito para classificar radiografia de tórax em 3 classes: atypical, indeterminate e typical para COVID19. Mais informações sobre o dataset estão disponíveis no arquivo README.md do diretório datasets.

O código que faz a conversão de .dcm para .png, que monta o dataframe com os pares imagem-classe, que faz a redestribuição do dataframe em treino e teste, que carrega essa distribuição em data generators e faz o fine-tuning da EfficientNetB5 está contido no diretório src.

Por favor, não esqueça de verificar a licensa deste projeto. Você também pode contribuir, basta sugerir modificações por um Pull Request.

About

Repositório utilizado para manter o projeto das discplinhas ECOM178 - REDES NEURAIS E APRENDIZADO PROFUNDO e ECOM181 - FUNDAMENTOS DE IA APLICADOS AO DIAGNÓSTICO MÉDICO.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published