BreastBox - ADetailer breast detection

세부 정보

파일 다운로드

모델 설명

BreastBox 검출 (YOLOv8s)

설명

BreastBoxADetailer를 사용하는 Stable Diffusion 사용자의 워크플로우를 개선하기 위해 특별히 훈련된 사용자 정의 객체 검출 모델로, YOLOv8s(yolo small) v2까지 지원합니다.

기본 모델들은 유방을 정확히 검출하지 못하거나, 측면 시점이나 특정 옷차림에서는 성능이 떨어지는 경우가 많지만, BreastBox는 판타지 캐릭터라도 보다 정밀하고 해부학적으로 정확한 바운딩 박스를 제공하는 것을 목표로 합니다. V1은 아직 완벽하지 않으며, 특히 특정 포즈나 옷 아래의 작은 유방에 대해서는 한계가 있지만, 향후 버전에서는 큰 개선이 있을 예정입니다.

이 모델을 사용하는 이유는?

  • 높은 정밀도: 수동으로 선별된 약 200장의 라벨링된 이미지(v1) 및 약 400장(v2)으로 훈련되었습니다.

  • 검출 능력: 측면 유방, 누워 있는 자세, 뒤에서의 시점, 부분적으로 가려진 경우 등 어려운 각도에서도 유방을 검출할 수 있습니다.

  • 스타일 무관: 주로 3DCG 이미지로 훈련되었지만, 사실적인/2D 생성물에도 어느 정도 일반화됩니다.

사용 방법 (ADetailer)

  1. .pt 파일을 다운로드합니다.

  2. ADetailer 모델 폴더에 배치합니다: \stable-diffusion-webui\models\ADetailer.

  3. ADetailer 탭에서 드롭다운 목록에서 이 모델을 선택합니다.

  4. 권장 설정:

    • 검출 신뢰도: 0.3 ~ 0.5 (검출이 놓치는 경우 조정하세요)

    • 마스크 병합 모드: Merge

버전 이력

  • v1.0 (실험적): 초기 릴리스. 표준 포즈와 대부분의 2DCG 스타일에서 매우 안정적인 성능을 보입니다. 극도로 작은 가슴이나 중력에 의해 형태가 크게 변형되는 “누워서 등을 대고, 거꾸로” 같은 포즈에서는 어려움을 겪을 수 있습니다 (~200개 데이터 샘플).

  • v2.0: 더 다양한 포즈, 유방 크기, 카메라 각도를 위한 데이터가 추가되어 더 우수한 일반화 성능을 제공합니다 (~400개 데이터 샘플).

  • v3.0 (진행 중): 손에 가려진 유방과 더 작은 유방에 대한 데이터를 추가하여 향상 중입니다 (~650개 데이터 샘플).

피드백

이 모델은 제 첫 번째 검출 훈련 프로젝트이며, 열정을 담아 제작되었습니다. 검출이 실패하는 특이 사례를 발견하시면, 댓글에 피드백이나 예시 이미지를 공유해 주세요. 다음 버전의 데이터셋 개선에 큰 도움이 됩니다!

이 모델로 만든 이미지

이미지를 찾을 수 없습니다.