From 10f6ac5e9b2ee69175e5cf4c356f3e77413ab9de Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 22 Nov 2023 21:16:55 +0100 Subject: [PATCH] `ultralytics 8.0.216` fix hard-coded `batch=64` cls loss (#6523) Signed-off-by: Glenn Jocher Co-authored-by: HDW AI group --- docs/hi/models/yolov3.md | 2 +- docs/hi/models/yolov5.md | 3 +-- docs/hi/models/yolov8.md | 2 +- docs/mkdocs_ar.yml | 11 +++++++++++ docs/mkdocs_de.yml | 11 +++++++++++ docs/mkdocs_es.yml | 11 +++++++++++ docs/mkdocs_fr.yml | 11 +++++++++++ docs/mkdocs_hi.yml | 11 +++++++++++ docs/mkdocs_ja.yml | 11 +++++++++++ docs/mkdocs_ko.yml | 11 +++++++++++ docs/mkdocs_pt.yml | 11 +++++++++++ docs/mkdocs_ru.yml | 11 +++++++++++ docs/mkdocs_zh.yml | 11 +++++++++++ ultralytics/__init__.py | 2 +- ultralytics/utils/loss.py | 2 +- 15 files changed, 115 insertions(+), 6 deletions(-) diff --git a/docs/hi/models/yolov3.md b/docs/hi/models/yolov3.md index 2a2c98f7a..4fc14a8c0 100644 --- a/docs/hi/models/yolov3.md +++ b/docs/hi/models/yolov3.md @@ -82,7 +82,7 @@ YOLOv3 श्रृंखला, इनमें YOLOv3, YOLOv3-Ultralytics औ अगर आप अपने शोध में YOLOv3 का उपयोग करते हैं, तो कृपया मूल YOLO पेपर्स और Ultralytics YOLOv3 रिपॉज़िटरी को उद्धृत करें। -!!! उध्दरण "" +!!! Quote "" === "BibTeX" diff --git a/docs/hi/models/yolov5.md b/docs/hi/models/yolov5.md index b1bcad567..4a9172e39 100644 --- a/docs/hi/models/yolov5.md +++ b/docs/hi/models/yolov5.md @@ -72,8 +72,7 @@ YOLOv5u वस्तु ज्ञापन के तरीकों में model.info() # COCO8 प्रायोगिक उदाहरण डेटासेट पर 100 एपॉक के लिए मॉडल - -ka प्रशिक्षित करें results = model.train(data='coco8.yaml', epochs=100, imgsz=640) + results = model.train(data='coco8.yaml', epochs=100, imgsz=640) # YOLOv5n मॉडल के साथ 'bus.jpg' छविमें ज्ञापन चलाएं results = model('path/to/bus.jpg') diff --git a/docs/hi/models/yolov8.md b/docs/hi/models/yolov8.md index a42debfd3..44f462804 100644 --- a/docs/hi/models/yolov8.md +++ b/docs/hi/models/yolov8.md @@ -36,7 +36,7 @@ Yएक मॉडल के हर मानक, विशिष्ट कार ## प्रदर्शन की मापदंड -!!! प्रदर्शन +!!! Note "प्रदर्शन" === "वस्तुनिर्धारण (COCO)" diff --git a/docs/mkdocs_ar.yml b/docs/mkdocs_ar.yml index c14ea8e37..966e1965b 100644 --- a/docs/mkdocs_ar.yml +++ b/docs/mkdocs_ar.yml @@ -188,6 +188,17 @@ nav: - الوضعية: tasks/pose.md - النماذج: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - المجموعات البيانية: - datasets/index.md diff --git a/docs/mkdocs_de.yml b/docs/mkdocs_de.yml index 0d01cc124..c63c47d41 100644 --- a/docs/mkdocs_de.yml +++ b/docs/mkdocs_de.yml @@ -188,6 +188,17 @@ nav: - Pose: tasks/pose.md - Modelle: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - Datensätze: - datasets/index.md diff --git a/docs/mkdocs_es.yml b/docs/mkdocs_es.yml index aad2873db..ec29a506e 100644 --- a/docs/mkdocs_es.yml +++ b/docs/mkdocs_es.yml @@ -188,6 +188,17 @@ nav: - Pose: tasks/pose.md - Modelos: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - Conjuntos de datos: - datasets/index.md diff --git a/docs/mkdocs_fr.yml b/docs/mkdocs_fr.yml index 65aacec6b..37f99d810 100644 --- a/docs/mkdocs_fr.yml +++ b/docs/mkdocs_fr.yml @@ -188,6 +188,17 @@ nav: - Pose: tasks/pose.md - Modèles: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - Jeux de données: - datasets/index.md diff --git a/docs/mkdocs_hi.yml b/docs/mkdocs_hi.yml index f82b3c9d4..45314bb42 100644 --- a/docs/mkdocs_hi.yml +++ b/docs/mkdocs_hi.yml @@ -188,6 +188,17 @@ nav: - मुद्रा: tasks/pose.md - मॉडल: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - डेटासेट्स: - datasets/index.md diff --git a/docs/mkdocs_ja.yml b/docs/mkdocs_ja.yml index 78b682cfd..53bd90667 100644 --- a/docs/mkdocs_ja.yml +++ b/docs/mkdocs_ja.yml @@ -188,6 +188,17 @@ nav: - ポーズ: tasks/pose.md - モデル: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - データセット: - datasets/index.md diff --git a/docs/mkdocs_ko.yml b/docs/mkdocs_ko.yml index 27471a20b..649bbc8db 100644 --- a/docs/mkdocs_ko.yml +++ b/docs/mkdocs_ko.yml @@ -188,6 +188,17 @@ nav: - 포즈: tasks/pose.md - 모델: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - 데이터셋: - datasets/index.md diff --git a/docs/mkdocs_pt.yml b/docs/mkdocs_pt.yml index 46979a489..724afabee 100644 --- a/docs/mkdocs_pt.yml +++ b/docs/mkdocs_pt.yml @@ -188,6 +188,17 @@ nav: - Pose: tasks/pose.md - Modelos: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - Conjuntos de Dados: - datasets/index.md diff --git a/docs/mkdocs_ru.yml b/docs/mkdocs_ru.yml index dd1a5bf53..b097c13d8 100644 --- a/docs/mkdocs_ru.yml +++ b/docs/mkdocs_ru.yml @@ -188,6 +188,17 @@ nav: - Поза: tasks/pose.md - Модели: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - Данные: - datasets/index.md diff --git a/docs/mkdocs_zh.yml b/docs/mkdocs_zh.yml index b96c69334..393d74979 100644 --- a/docs/mkdocs_zh.yml +++ b/docs/mkdocs_zh.yml @@ -188,6 +188,17 @@ nav: - 姿态: tasks/pose.md - 模型: - models/index.md + - YOLOv3: models/yolov3.md + - YOLOv4: models/yolov4.md + - YOLOv5: models/yolov5.md + - YOLOv6: models/yolov6.md + - YOLOv7: models/yolov7.md + - YOLOv8: models/yolov8.md + - SAM (Segment Anything Model): models/sam.md + - MobileSAM (Mobile Segment Anything Model): models/mobile-sam.md + - FastSAM (Fast Segment Anything Model): models/fast-sam.md + - YOLO-NAS (Neural Architecture Search): models/yolo-nas.md + - RT-DETR (Realtime Detection Transformer): models/rtdetr.md - 数据集: - datasets/index.md diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index 3469720c0..8686d6a49 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.215' +__version__ = '8.0.216' from ultralytics.models import RTDETR, SAM, YOLO from ultralytics.models.fastsam import FastSAM diff --git a/ultralytics/utils/loss.py b/ultralytics/utils/loss.py index 916f91c56..da2e58423 100644 --- a/ultralytics/utils/loss.py +++ b/ultralytics/utils/loss.py @@ -523,6 +523,6 @@ class v8ClassificationLoss: def __call__(self, preds, batch): """Compute the classification loss between predictions and true labels.""" - loss = torch.nn.functional.cross_entropy(preds, batch['cls'], reduction='sum') / 64 + loss = torch.nn.functional.cross_entropy(preds, batch['cls'], reduction='mean') loss_items = loss.detach() return loss, loss_items