12345AILI/image-dataset-sorter

O que image-dataset-sorter faz?

Skill para Cursor Agent: Classifique e organize rapidamente imagens do GPT-Image2 / Banana, arquivando-as por nível de qualidade, para criar conjuntos de dados de treinamento de IA.

Funcionalidades

• Classificação automática: Separa imagens em pastas good/, mid/ e bad/ com base em heurísticas de qualidade (nitidez, ruído, composição).
• Suporte a GPT-Image2 e Banana: Funciona com os formatos de saída padrão dessas plataformas.
• Organização em lote: Processa múltiplas imagens de uma vez, criando uma estrutura de diretórios limpa.
• Preparação para dataset: Gera um arquivo dataset.csv com metadados e classificação de cada imagem.

Como usar

1. Execute o comando no Cursor Agent:

   cursor agent --skill "12345AILI/image-dataset-sorter" --args "--input <pasta_imagens>"
   

2. O skill analisará cada imagem e a moverá para a subpasta correspondente.
3. O resultado será exibido no terminal, com contagem de imagens por categoria.

Configuração

• Limiar de qualidade: Ajustável via argumento --threshold (padrão: 0.5).
• Formato de saída: Suporta .png, .jpg, .jpeg, .webp.
• Exclusão de duplicatas: Ativada por padrão (--dedupe).

Exemplo

cursor agent --skill "12345AILI/image-dataset-sorter" --args "--input ./gpt_images --threshold 0.6 --dedupe"

Notas

• Certifique-se de ter permissão de leitura e escrita nas pastas.
• Imagens classificadas como "bad" são movidas para bad/ e podem ser revisadas manualmente.
• Este skill é ideal para preparar dados para fine-tuning de modelos de imagem.