Customizable neural network plotter
Com esse script, você pode criar uma representação gráfica de sua rede neural, com os neurônios, camadas e ligações entre camadas. O objetivo do projeto é auxiliar em uma rápida criação de imagens assim para maior agilidade em criação de documentações de projetos que as usam. O funcionamento é relativamente simples, onde se usa a biblioteca networkx para criação do grafo, e cálculos simples para criar posições adequadas para cada neurônio. A imagem é por fim passada para um objeto matplotlib e salva na pasta de images com um nome descritivo com base na quantidade de neurônios de cada camada, evitando que as imagens anteriores sejam sobrescritas.
Para utilizar este projeto, siga os passos abaixo:
- Clone o Repositório: Para começar, clone o repositório do projeto para o seu sistema local. Use o seguinte comando no terminal:
git clone https://github.com/Daniel-Alvarenga/Neural_Netwok_Plotter
- Navegue para a Pasta do Projeto: Navegue para a pasta do projeto usando o comando
cd
. Por exemplo:
cd Neural_Netwok_Plotter
- Crie um Ambiente Virtual (Opcional): Recomenda-se criar um ambiente virtual para isolar as dependências deste projeto. Use
venv
(Python 3) ouvirtualenv
:
python -m venv venv
Ative o ambiente virtual:
- No Windows:
venv\Scripts\activate
- No macOS e Linux:
source venv/bin/activate
- Instale as Dependências: Certifique-se de que as dependências necessárias estejam instaladas. Você pode usar o
requirements.txt
fornecido para instalar todas as dependências de uma vez:
pip install -r requirements.txt
- Execute o
load.py
: Após configurar o ambiente e instalar as dependências, você pode executar oload.py
para carregar o modelo treinado e reconhecer dígitos em novas imagens. Use o seguinte comando:python plotter.py
Siga as instruções para que o script tenha os dados de sua rede. Ao finalizar entrando com -1, você terá a imagem salva na pasta image.