VIGIS
Um sistema inteligente e completo para coleta, treinamento e classificação de imagens, com explicabilidade visual via Grad-CAM.
O Pipeline Completo
1. Colete seus Dados
Baixe centenas de imagens de qualquer categoria automaticamente usando a busca do Bing.
python download_images.py --classe "foguete"
2. Treine seu Modelo
Use os dados coletados para treinar uma Rede Neural Convolucional (CNN) personalizada com um único comando.
python train.py
3. Classifique e Entenda
Classifique novas imagens e visualize as áreas que o modelo usou para tomar sua decisão com Grad-CAM.
python classify.py --image "img.jpg" --gradcam
Explicabilidade com Grad-CAM
Entenda o "Porquê"
VIGIS não apenas classifica suas imagens, ele mostra onde o modelo está "olhando" para tomar uma decisão. A técnica Grad-CAM gera um "mapa de calor" visual, destacando as regiões mais importantes da imagem para a classificação, o que é crucial para depurar e confiar no seu modelo.
Imagem Original
Com Grad-CAM
Comece em Minutos
# 1. Clone o repositório
git clone https://github.com/SyraDevOps/Vigilis.git
cd Vigilis
# 2. Instale as dependências
pip install torch torchvision opencv-python icrawler
# 3. Baixe imagens para uma classe (ex: gatos)
python download_images.py --classe "gatos" --quantidade 150
# 4. Treine o modelo com os dados baixados
python train.py
# 5. Classifique uma nova imagem e veja a mágica
python classify.py --image "caminho/para/imagem_de_gato.jpg" --gradcam