Модель YOLO11n (nano), обученная на 367 фотографиях реальных женщин в различных позах, разных рас, в разной обстановке. Детектирует один класс — грудь (женская, обнажённая).
Основная цель проекта — учебная: показать полный цикл обучения YOLO на собственном датасете с нуля:
- Сбор и разметка данных (367 изображений)
- Подготовка датасета в формате YOLO
- Обучение модели (YOLO11nano)
- Инференс и валидация
# Установка
pip install ultralytics
# Инференс (предобученная модель)
from ultralytics import YOLO
model = YOLO("titsyolo11n.pt")
results = model("image.jpg")
results[0].show()| Файл | Описание |
|---|---|
titsyolo11n.pt |
Предобученная модель YOLO11n (веса) |
from ultralytics import YOLO
model = YOLO("yolo11n.pt") # предобученная baseline
model.train(
data="dataset.yaml",
epochs=100,
imgsz=640,
batch=16,
name="tits_yolo11n",
)| Параметр | Значение |
|---|---|
| Архитектура | YOLO11nano |
| Классов | 1 (breast) |
| Датасет | 367 изображений (обнажённая женская грудь, в репозитории отсутствует) |
| Размер входа | 640×640 |
| Формат весов | PyTorch (.pt) |
- Проект создан в учебных целях для демонстрации пайплайна тренировки кастомной YOLO-модели
- Модель обучена на изображениях обнажённой женской груди (NSFW-контент, в репозитории отсутствует)
- Малый датасет (367 фото) не подходит для production-использования
- Для серьёзных задач необходимы тысячи размеченных изображений
Учебный проект для освоения YOLO и computer vision