{"id":25165581,"url":"https://github.com/pauloruszel/yolo11_face_detection","last_synced_at":"2026-05-05T21:34:45.025Z","repository":{"id":276369029,"uuid":"927971870","full_name":"pauloruszel/yolo11_face_detection","owner":"pauloruszel","description":null,"archived":false,"fork":false,"pushed_at":"2025-02-07T19:49:39.000Z","size":22040,"stargazers_count":0,"open_issues_count":0,"forks_count":0,"subscribers_count":1,"default_branch":"master","last_synced_at":"2025-04-03T16:42:56.430Z","etag":null,"topics":["cuda","nvcc","nvidia-gpu","pip","python3","pytorch","widerface-dataset","yolo11"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/pauloruszel.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":null,"code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2025-02-05T21:00:34.000Z","updated_at":"2025-02-07T19:49:42.000Z","dependencies_parsed_at":"2025-02-07T20:40:36.508Z","dependency_job_id":null,"html_url":"https://github.com/pauloruszel/yolo11_face_detection","commit_stats":null,"previous_names":["pauloruszel/yolo11_face_detection"],"tags_count":0,"template":false,"template_full_name":null,"purl":"pkg:github/pauloruszel/yolo11_face_detection","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/pauloruszel%2Fyolo11_face_detection","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/pauloruszel%2Fyolo11_face_detection/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/pauloruszel%2Fyolo11_face_detection/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/pauloruszel%2Fyolo11_face_detection/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/pauloruszel","download_url":"https://codeload.github.com/pauloruszel/yolo11_face_detection/tar.gz/refs/heads/master","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/pauloruszel%2Fyolo11_face_detection/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":32669431,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-05-05T11:29:49.557Z","status":"ssl_error","status_checked_at":"2026-05-05T11:29:48.587Z","response_time":54,"last_error":"SSL_connect returned=1 errno=0 peeraddr=140.82.121.5:443 state=error: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["cuda","nvcc","nvidia-gpu","pip","python3","pytorch","widerface-dataset","yolo11"],"created_at":"2025-02-09T05:26:46.281Z","updated_at":"2026-05-05T21:34:45.002Z","avatar_url":"https://github.com/pauloruszel.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# YOLO11 Face Detection - Guia de Uso\r\n\r\nEste repositório fornece scripts para treinamento e inferência de um modelo YOLO para detecção de rostos.\r\n\r\n---\r\n\r\n## Requisitos\r\n\r\nAntes de iniciar, certifique-se de ter as seguintes versões instaladas:\r\n\r\n### Python e Pip\r\n\r\n- **Python**: 3.12.7\r\n- **Pip**: 25.0\r\n\r\nComandos para verificação:\r\n\r\n```sh\r\npython --version\r\npip --version\r\n```\r\n\r\n### PyTorch e CUDA\r\n\r\n- **PyTorch**: 2.5.1+cu121\r\n- **CUDA**: 12.1\r\n- **NVCC (CUDA Compiler)**: 12.1.66\r\n\r\nComandos para verificação:\r\n\r\n```sh\r\npython -c \"import torch; print(torch.__version__); print(torch.version.cuda)\"\r\nnvcc --version\r\n```\r\n\r\n---\r\n\r\n## Instalação do Ambiente\r\n\r\n### 1. Instalação do Python e Pip\r\n\r\nBaixe e instale o **Python 3.12.7** a partir do site oficial:\r\n\r\n- [Download Python](https://www.python.org/downloads/release/python-3127/)\r\n\r\n### 2. Instalação do CUDA e CuDNN\r\n\r\nBaixe e instale as versões compatíveis do CUDA e CuDNN:\r\n\r\n- **CUDA 12.1**: [Download CUDA](https://developer.nvidia.com/cuda-12-1-download-archive)\r\n- **CuDNN para CUDA 12.1**: [Download CuDNN](https://developer.nvidia.com/cudnn)\r\n\r\n### 3. Instalação do PyTorch\r\n\r\nInstale a versão correta do PyTorch com suporte ao CUDA 12.1:\r\n\r\n```sh\r\npip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121\r\n```\r\n\r\n### 4. Clonar o Repositório e Criar o Ambiente Virtual\r\n\r\n```sh\r\ngit clone https://github.com/pauloruszel/yolo11_face_detection.git\r\ncd yolo11_face_detection\r\npython -m venv venv\r\nsource venv/bin/activate # Linux/Mac\r\nvenv\\Scripts\\activate  # Windows\r\n```\r\n\r\n---\r\n\r\n## Execução dos Scripts\r\n\r\n### 1. Converter as Labels para YOLO\r\n\r\nEste script converte as anotações do dataset WIDER FACE para o formato do YOLO.\r\n\r\n```sh\r\npython scripts/convert_wider_to_yolo.py\r\n```\r\n\r\n### 2. Corrigir Labels Inconsistentes\r\n\r\nEste script verifica e corrige labels que possuam coordenadas fora dos limites permitidos pelo YOLO.\r\n\r\n```sh\r\npython scripts/fix_labels.py\r\n```\r\n\r\n### 3. Treinar o Modelo\r\n\r\nO batch size será ajustado automaticamente com base na memória disponível da GPU.\r\n\r\n- `--img 640`: Define o tamanho da imagem utilizada no treinamento.\r\n- `--batch 4`: Define o tamanho do batch (pode ser ajustado automaticamente).\r\n- `--epochs 100`: Quantidade de épocas de treinamento.\r\n- `--data dataset.yaml`: Caminho para o arquivo de configuração do dataset.\r\n- `--weights yolo11n.pt`: Modelo pré-treinado utilizado como base.\r\n- `--device 0`: Define a GPU a ser utilizada (0 para a primeira GPU).\r\n- `--name face_detection`: Nome do experimento.\r\n- `--half`: Usa precision FP16 para reduzir consumo de memória.\r\n- `--cache=ram`: Carrega as imagens na RAM para acelerar o treinamento.\r\n\r\n```sh\r\npython scripts/train.py --img 640 --batch 4 --epochs 100 --data \"C:/caminho/yolo11_face_detection/dataset.yaml\" --weights \"C:/caminho/yolo11_face_detection/models/yolo11n.pt\" --device 0 --name face_detection --half --cache=ram\r\n```\r\n\r\n### 4. Executar Inferência com Webcam\r\n\r\nEste script utiliza a webcam para realizar a detecção em tempo real.\r\n\r\n```sh\r\npython scripts/webcam_detect.py\r\n```\r\n\r\n### 5. Testar o Modelo em uma Imagem\r\n\r\nEste script aplica a detecção em uma imagem específica.\r\n\r\n```sh\r\npython scripts/detection.py --image datasets/images/val/alguma_imagem.jpg\r\n```\r\n\r\n---\r\n\r\n## Estrutura do Diretório\r\n\r\n```\r\n.\r\n├── datasets/\r\n│   ├── images/\r\n│   │   ├── train/\r\n│   │   ├── val/\r\n│   ├── labels/\r\n│       ├── train/\r\n│       ├── val/\r\n├── models/\r\n├── scripts/\r\n│   ├── train.py\r\n│   ├── webcam_detect.py\r\n│   ├── detection.py\r\n│   ├── convert_wider_to_yolo.py\r\n│   ├── fix_labels.py\r\n├── yolo11n.pt\r\n└── README.md\r\n```\r\n\r\n---\r\n\r\n## Dicas de Otimização\r\n\r\n- **Ajuste o Batch Size**: O batch size é ajustado automaticamente para evitar **out of memory (OOM)** na GPU.\r\n- **Utilize `--cache=ram`**: Acelera o treinamento carregando as imagens na memória RAM.\r\n- **Monitore a GPU**:\r\n  ```sh\r\n  nvidia-smi\r\n  ```\r\n- **Caso encontre erros de CUDA**, tente reiniciar a GPU com:\r\n  ```sh\r\n  nvidia-smi --gpu-reset\r\n  ```\r\n\r\n---\r\n\r\nCaso encontre problemas, verifique se todas as versões dos pacotes estão compatíveis e atualizadas. Boa sorte no treinamento! 🚀\r\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fpauloruszel%2Fyolo11_face_detection","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fpauloruszel%2Fyolo11_face_detection","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fpauloruszel%2Fyolo11_face_detection/lists"}