簡介本資源是面向人工智能與計算機視覺初學(xué)者及算法工程師的吸煙行為識別專用標(biāo)注數(shù)據(jù)集適用于圖像分類、目標(biāo)檢測模型訓(xùn)練與健康行為分析等AI項目開發(fā)。數(shù)據(jù)集共1567個文件包含783張JPG格式吸煙場景實拍圖像、783份對應(yīng)XML標(biāo)注文件含邊界框與類別標(biāo)簽以及1份說明文檔TXT總大小31.71MB圖像命名統(tǒng)一規(guī)范如smoke_a273.jpg便于批量加載與路徑解析XML標(biāo)注結(jié)構(gòu)清晰支持Pascal VOC格式直接導(dǎo)入主流框架。目前已有158人學(xué)習(xí)下載體現(xiàn)了該數(shù)據(jù)集在輕量級健康A(chǔ)I應(yīng)用中的實際需求。用戶可直接用于構(gòu)建吸煙/非吸煙二分類模型或擴展為多任務(wù)學(xué)習(xí)配套標(biāo)注信息完整覆蓋人物姿態(tài)、手持動作及環(huán)境上下文顯著降低數(shù)據(jù)預(yù)處理門檻同時兼顧隱私合規(guī)性設(shè)計適合教學(xué)實驗、課程設(shè)計及科研原型驗證。1. 項目概述從“吸煙數(shù)據(jù)集帶標(biāo)注”說起最近在做一個關(guān)于公共安全或者健康監(jiān)測相關(guān)的項目需要訓(xùn)練一個能識別吸煙行為的模型。找了一圈公開數(shù)據(jù)集發(fā)現(xiàn)直接可用的、標(biāo)注質(zhì)量高的“吸煙行為”數(shù)據(jù)集并不多。很多朋友可能也遇到過類似的問題想做一個特定的視覺識別任務(wù)但第一步——找數(shù)據(jù)——就卡住了。要么數(shù)據(jù)量不夠要么標(biāo)注不規(guī)范要么場景太單一。所以我決定自己動手整理和構(gòu)建一個專注于“吸煙”行為的圖像/視頻數(shù)據(jù)集并完成高質(zhì)量的標(biāo)注。這不僅僅是一個數(shù)據(jù)包的集合更涉及到數(shù)據(jù)采集的策略、標(biāo)注規(guī)范的制定、工具的選擇以及質(zhì)量控制的完整流程。如果你正在研究行為識別、異常檢測或者單純想學(xué)習(xí)如何從零開始構(gòu)建一個定制化的數(shù)據(jù)集那么我踩過的這些坑和總結(jié)的經(jīng)驗或許能給你省下不少時間。2. 數(shù)據(jù)集構(gòu)建的核心思路與設(shè)計考量2.1 明確任務(wù)定義與數(shù)據(jù)需求構(gòu)建數(shù)據(jù)集的第一步絕不是馬上開始收集圖片而是想清楚你的模型最終要解決什么問題。對于“吸煙識別”這個任務(wù)我們需要進一步細化識別目標(biāo)是什么是識別“香煙”這個物體還是識別“人正在吸煙”這個動作這兩者有本質(zhì)區(qū)別。前者是靜態(tài)目標(biāo)檢測后者是動態(tài)行為識別可能涉及時序信息。更復(fù)雜的可能是要識別“在禁煙區(qū)域吸煙”的違規(guī)行為這就需要同時檢測“人”、“香煙”和“場景標(biāo)識”。應(yīng)用場景是什么是室內(nèi)監(jiān)控攝像頭下的實時檢測還是對網(wǎng)絡(luò)圖片的離線審核不同的場景對數(shù)據(jù)的多樣性要求不同。監(jiān)控場景可能光線多變、角度固定、人物較小網(wǎng)絡(luò)圖片則背景復(fù)雜、姿態(tài)多樣、圖片質(zhì)量不一。需要怎樣的標(biāo)注基于任務(wù)定義我們確定標(biāo)注類型。常見的有邊界框Bounding Box標(biāo)注香煙、手持香煙的手部或者吸煙者的頭部/上半身。這是最常用的適用于YOLO、Faster R-CNN等目標(biāo)檢測模型。關(guān)鍵點Keypoints標(biāo)注手部關(guān)節(jié)、香煙末端、嘴部等關(guān)鍵點。這對于精確判斷“手持”和“靠近嘴部”的動作關(guān)聯(lián)很有用常用于姿態(tài)估計結(jié)合行為識別。實例分割I(lǐng)nstance Segmentation精確勾勒出香煙和手部的輪廓。標(biāo)注成本高但精度也更高。行為標(biāo)簽Action Label為一段視頻或一系列圖片打上“吸煙/非吸煙”的標(biāo)簽。這通常用于視頻分類或時序動作檢測模型。我的核心需求是做一個通用的、可用于訓(xùn)練高性能目標(biāo)檢測模型如YOLOv8的數(shù)據(jù)集因此選擇邊界框標(biāo)注作為主要形式標(biāo)注對象為“香煙”包括被手持的、放在桌上的、嘴里的和“吸煙者”當(dāng)香煙與人體關(guān)聯(lián)度很高時標(biāo)注人可能有助于上下文判斷。同時為每張圖片附加一個“是否包含吸煙行為”的分類標(biāo)簽。2.2 數(shù)據(jù)采集策略與來源規(guī)劃高質(zhì)量的數(shù)據(jù)集需要多樣性和代表性。我規(guī)劃了以下幾個數(shù)據(jù)來源以覆蓋盡可能多的場景公開數(shù)據(jù)集挖掘與篩選這是起點。我會搜索包含“吸煙”、“smoking”、“cigarette”等標(biāo)簽的現(xiàn)有數(shù)據(jù)集例如某些大型通用物體檢測數(shù)據(jù)集中的相關(guān)子集或者安全監(jiān)控、零售行為分析數(shù)據(jù)集。但需要仔細清洗確保標(biāo)注準(zhǔn)確。網(wǎng)絡(luò)爬蟲合規(guī)前提下在遵守版權(quán)和隱私法規(guī)的前提下從一些公開的圖片分享平臺、視頻網(wǎng)站截取關(guān)鍵幀獲取數(shù)據(jù)。關(guān)鍵詞需要精心設(shè)計如“man smoking”、“woman with cigarette”、“smoking at table”、“smoking outdoors night”等以覆蓋不同性別、場景、光照和時間。模擬場景拍攝這是保證數(shù)據(jù)質(zhì)量和特定場景覆蓋的關(guān)鍵。在獲得授權(quán)和確保安全的情況下可以邀請志愿者在多種背景辦公室、樓梯間、公園、車內(nèi)下模擬各種吸煙姿態(tài)手持、口含、點火、彈煙灰進行拍攝。這能有效補充網(wǎng)絡(luò)數(shù)據(jù)中難以找到的清晰、多角度的正面樣本。合成數(shù)據(jù)生成作為補充可以使用3D渲染或圖像合成技術(shù)如將香煙的3D模型合成到不同背景和手持姿態(tài)的圖片中。這在數(shù)據(jù)稀缺的初期或需要極端視角時非常有用但要注意“模擬到真實”的域適應(yīng)問題。注意無論哪種來源都必須高度重視隱私和倫理。對所有人臉、車牌等敏感信息進行模糊化處理。模擬拍攝需獲得參與者知情同意。合成數(shù)據(jù)則避免了隱私問題。2.3 標(biāo)注規(guī)范制定統(tǒng)一標(biāo)準(zhǔn)是質(zhì)量的基石這是最容易被忽視卻至關(guān)重要的一步。如果標(biāo)注標(biāo)準(zhǔn)模糊后期模型訓(xùn)練會混亂不堪。我制定的規(guī)范如下“香煙”物體標(biāo)注邊界框緊密度框體應(yīng)緊密貼合香煙的可見部分包括濾嘴和煙體但不必包含煙霧。對于被手指部分遮擋的香煙按可見部分標(biāo)注。遮擋處理香煙被完全遮擋超過50%則不予標(biāo)注視為不可識別。部分遮擋時標(biāo)注可見部分。角度與形態(tài)無論是側(cè)面、端頭視角還是彎曲、折斷狀態(tài)只要可辨識為香煙均需標(biāo)注。相似物區(qū)分特別注意與筆、棍狀糖果、溫度計等細長物體的區(qū)分。僅標(biāo)注確認(rèn)為香煙的物體。“吸煙者”標(biāo)注可選當(dāng)香煙明顯與人的手部或嘴部接觸構(gòu)成吸煙動作時建議同時標(biāo)注該人物通常用一個較大的框框住上半身或全身。這有助于模型學(xué)習(xí)“吸煙”這個行為的上下文而不僅僅是孤立地識別香煙。圖像級標(biāo)簽smoking圖片中包含正在進行的吸煙行為香煙與口/手接觸。cigarette_present圖片中包含香煙但可能未被使用如放在桌上。non_smoking圖片中無任何香煙或吸煙行為。標(biāo)注格式選擇與未來訓(xùn)練框架兼容的格式。我選擇YOLO格式因為它簡單且被廣泛支持。每張圖片對應(yīng)一個.txt文件每行內(nèi)容為class_id x_center y_center width height坐標(biāo)均為歸一化后的值0-1之間。3. 數(shù)據(jù)標(biāo)注實操工具、流程與質(zhì)量控制3.1 標(biāo)注工具選型與配置工欲善其事必先利其器。我對比了幾款主流開源標(biāo)注工具LabelImg老牌經(jīng)典支持矩形框標(biāo)注輸出PASCAL VOC或YOLO格式。界面簡單但對于大規(guī)模數(shù)據(jù)集效率較低。Labelme由麻省理工開發(fā)功能強大支持多邊形、矩形、圓形、線段、點等多種標(biāo)注輸出JSON格式。非常適合需要分割或關(guān)鍵點標(biāo)注的任務(wù)但處理純矩形框標(biāo)注時操作略顯繁瑣。CVAT英特爾開發(fā)的在線視頻/圖像標(biāo)注工具功能極其強大支持團隊協(xié)作、自動標(biāo)注、視頻插幀、多種任務(wù)類型。是工業(yè)級項目的首選但部署和上手有一定復(fù)雜度。Makesense.ai一個免費的在線標(biāo)注工具無需安裝支持拖拽上傳、自動標(biāo)注建議集成YOLO等模型進行預(yù)標(biāo)注、團隊協(xié)作。對于個人或小團隊快速啟動項目非常友好。我的選擇與理由考慮到項目初期需要快速迭代和驗證且以矩形框標(biāo)注為主我最終選擇了Makesense.ai。它的優(yōu)勢在于零安裝打開瀏覽器就能用避免了環(huán)境配置問題。預(yù)標(biāo)注功能可以上傳一個預(yù)訓(xùn)練的權(quán)重哪怕是通用模型如YOLOv8s讓它先跑一遍人工只需修正和補充能極大提升效率尤其是在標(biāo)注“香煙”這種特征相對明顯的物體時。格式支持全直接導(dǎo)出YOLO、PASCAL VOC、COCO等多種格式。協(xié)作方便可以生成項目鏈接邀請他人共同標(biāo)注。對于需要更復(fù)雜標(biāo)注如關(guān)鍵點或處理視頻流的后期階段CVAT是更專業(yè)的選擇。3.2 標(biāo)準(zhǔn)化標(biāo)注流程SOP為了確保多人協(xié)作時質(zhì)量一致我制定了詳細的標(biāo)注操作流程數(shù)據(jù)預(yù)處理與導(dǎo)入將所有收集到的圖片統(tǒng)一縮放至一個標(biāo)準(zhǔn)尺寸如1333x800保持長寬比多余部分填充灰邊以減少后續(xù)模型輸入尺寸不一帶來的影響。按8:1:1的比例隨機劃分訓(xùn)練集、驗證集和測試集并在項目開始時就固定下來。將劃分好的圖片文件夾上傳至Makesense.ai創(chuàng)建新項目。標(biāo)注任務(wù)分配與界面熟悉定義標(biāo)簽類別cigarette,smoker。為標(biāo)注人員提供詳細的標(biāo)注規(guī)范文檔和帶示例的圖解。讓每位標(biāo)注員先標(biāo)注同一批如20張校準(zhǔn)圖片由負責(zé)人檢查并統(tǒng)一反饋確保所有人理解一致。核心標(biāo)注操作使用矩形框工具緊密框選目標(biāo)。對于模糊、存疑的樣本標(biāo)記為“困難樣本”或放入待討論隊列而不是隨意標(biāo)注或忽略。充分利用快捷鍵如Makesense中按Ctrl鍵可吸附邊緣提升速度。利用預(yù)標(biāo)注加速在標(biāo)注一部分?jǐn)?shù)據(jù)如200張后訓(xùn)練一個初版的YOLOv8模型。將這個模型權(quán)重用于后續(xù)批次的預(yù)標(biāo)注。標(biāo)注員的工作從“從零開始畫框”變?yōu)椤皩徍撕托拚Ｐ皖A(yù)測的框”效率可提升50%以上。定期導(dǎo)出與備份每完成一個批次如500張立即導(dǎo)出YOLO格式的標(biāo)注文件并與原圖一起備份。備份時保持圖片和標(biāo)注文件的目錄結(jié)構(gòu)一致。3.3 質(zhì)量控制與迭代優(yōu)化標(biāo)注質(zhì)量直接決定模型天花板。我建立了三層質(zhì)檢機制標(biāo)注員自檢標(biāo)注完成后強制要求回看一遍自己標(biāo)注的所有框檢查是否有漏標(biāo)、錯標(biāo)、框體質(zhì)量差的問題。交叉互檢標(biāo)注員之間隨機交換一部分已標(biāo)注數(shù)據(jù)進行檢查。不同的人更容易發(fā)現(xiàn)對方習(xí)慣性忽略的錯誤。負責(zé)人抽檢與仲裁項目負責(zé)人隨機抽取至少10%的數(shù)據(jù)進行詳細檢查。對于交叉互檢中爭議的樣本由負責(zé)人根據(jù)規(guī)范最終裁定。常見的質(zhì)量問題及處理框體過松或過緊通過校準(zhǔn)圖片和示例反復(fù)強調(diào)“緊密貼合”的原則。類別混淆明確“香煙”與“類似物”的區(qū)分標(biāo)準(zhǔn)制作對比圖例。遮擋與截斷處理不一致統(tǒng)一規(guī)則被遮擋超過50%不標(biāo)截斷的如圖片邊緣標(biāo)注可見部分。微小目標(biāo)漏標(biāo)遠處的、像素很小的香煙是難點。要求標(biāo)注員放大圖片仔細查看對于實在難以確認(rèn)的標(biāo)記為“疑似”由負責(zé)人用更高分辨率原圖判斷。質(zhì)檢發(fā)現(xiàn)的問題要記錄到“錯誤類型統(tǒng)計表”中定期回顧并更新標(biāo)注規(guī)范或?qū)?biāo)注員進行再培訓(xùn)。這是一個持續(xù)迭代的過程。4. 數(shù)據(jù)集整理與YOLOv8訓(xùn)練準(zhǔn)備4.1 數(shù)據(jù)格式整理與增強標(biāo)注完成后我們得到的是圖片文件和對應(yīng)的.txt標(biāo)注文件。為了適配YOLOv8的訓(xùn)練需要整理成固定的結(jié)構(gòu)Smoking_Dataset/ ├── images/ │ ├── train/ │ │ ├── image_001.jpg │ │ └── ... │ ├── val/ │ │ ├── image_101.jpg │ │ └── ... │ └── test/ │ ├── image_201.jpg │ └── ... └── labels/ ├── train/ │ ├── image_001.txt │ └── ... ├── val/ │ ├── image_101.txt │ └── ... └── test/ ├── image_201.txt └── ...接下來分析數(shù)據(jù)集的基本情況這對于后續(xù)訓(xùn)練策略調(diào)整至關(guān)重要。可以使用Python腳本快速統(tǒng)計圖像數(shù)量與分辨率分布訓(xùn)練集、驗證集、測試集各多少張圖片大多是高分辨率還是低分辨率目標(biāo)數(shù)量與尺寸分布統(tǒng)計每個類別cigarette,smoker的實例總數(shù)。更重要的是分析目標(biāo)邊界框的寬度和高度歸一化后的分布。這能告訴你數(shù)據(jù)集中目標(biāo)大多是“小目標(biāo)”、“中目標(biāo)”還是“大目標(biāo)”。吸煙數(shù)據(jù)集里香煙往往是“小目標(biāo)”這會影響模型anchor的設(shè)計或訓(xùn)練時的關(guān)注度。類別平衡性檢查cigarette和smoker的實例數(shù)是否懸殊過大。如果嚴(yán)重不平衡需要考慮過采樣、欠采樣或給不同類別分配不同的損失權(quán)重?；诜治鼋Y(jié)果進行數(shù)據(jù)增強以增加多樣性并緩解過擬合。YOLOv8內(nèi)置了強大的增強管道但理解其原理有助于我們自定義。對于吸煙數(shù)據(jù)集我重點應(yīng)用以下增強幾何變換旋轉(zhuǎn)小角度如±15度因為香煙通常不會大角度傾斜、平移、縮放、剪切。模擬拍攝時的視角變化。色彩變換亮度、對比度、飽和度、色調(diào)的隨機調(diào)整。模擬不同光照條件室內(nèi)昏暗、室外強光?；旌项愒鰪奙osaic四圖拼接和MixUp能極大地豐富背景并讓模型學(xué)習(xí)在不同上下文和尺度中識別目標(biāo)對于小目標(biāo)檢測尤其有效。針對小目標(biāo)的增強隨機將訓(xùn)練集中的小目標(biāo)如香煙復(fù)制粘貼到圖像的其他位置需確保合理性如不會貼到天空增加小目標(biāo)的出現(xiàn)頻率。4.2 配置YOLOv8訓(xùn)練環(huán)境與參數(shù)解析環(huán)境配置相對簡單使用Conda創(chuàng)建虛擬環(huán)境安裝PyTorch和Ultralytics的YOLOv8包即可。conda create -n yolo_smoke python3.8 conda activate yolo_smoke pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 # 根據(jù)CUDA版本選擇 pip install ultralytics關(guān)鍵的一步是準(zhǔn)備數(shù)據(jù)集配置文件dataset.yaml。這個文件告訴YOLOv8你的數(shù)據(jù)在哪、有哪些類別。# dataset.yaml path: /path/to/Smoking_Dataset # 數(shù)據(jù)集根目錄 train: images/train # 訓(xùn)練集圖片相對路徑 val: images/val # 驗證集圖片相對路徑 test: images/test # 測試集圖片相對路徑可選 # 類別列表 names: 0: cigarette 1: smoker接下來是理解并調(diào)整訓(xùn)練參數(shù)。直接使用默認(rèn)參數(shù)可能有效但針對我們的數(shù)據(jù)集進行微調(diào)能獲得更好效果。以下是一些核心參數(shù)解析model: 選擇模型尺寸如yolov8n.pt納米、yolov8s.pt小、yolov8m.pt中。對于初步實驗從yolov8s開始是個好選擇在精度和速度間取得平衡。data: 指向上面創(chuàng)建的dataset.yaml。epochs: 訓(xùn)練輪數(shù)。對于中等規(guī)模數(shù)據(jù)集幾千張圖片100-300輪是常見的起點?？梢酝ㄟ^觀察驗證集損失曲線提前停止。imgsz: 輸入圖像尺寸。默認(rèn)640。如果原始圖像中目標(biāo)香煙非常小可以嘗試增大到832甚至1024但這會顯著增加顯存消耗和訓(xùn)練時間。batch: 批次大小。取決于你的GPU顯存。在顯存允許的情況下盡可能設(shè)大如16, 32這有助于訓(xùn)練穩(wěn)定。workers: 數(shù)據(jù)加載的進程數(shù)。通常設(shè)為CPU核心數(shù)可以加快數(shù)據(jù)讀取。lr0和lrf: 初始學(xué)習(xí)率和最終學(xué)習(xí)率因子。學(xué)習(xí)率是訓(xùn)練中最關(guān)鍵的參數(shù)之一。對于小數(shù)據(jù)集初始學(xué)習(xí)率lr0可以設(shè)小一點如0.01避免震蕩。lrf是最終學(xué)習(xí)率衰減到的比例。weight_decay: 權(quán)重衰減防止過擬合的正則化項。默認(rèn)值通?？尚?。box: 邊界框回歸損失的權(quán)重。對于檢測任務(wù)這是主要損失。cls: 分類損失的權(quán)重。如果分類任務(wù)困難如區(qū)分不同香煙品牌但我們不需要可以調(diào)整。我們只有兩個簡單類別默認(rèn)即可。dfl: Distribution Focal Loss的權(quán)重用于優(yōu)化邊界框分布。YOLOv8的新特性通常保持默認(rèn)。我的經(jīng)驗是對于自定義數(shù)據(jù)集最需要關(guān)注的參數(shù)是imgsz、batch、lr0和epochs?？梢韵仁褂媚J(rèn)參數(shù)跑一個基線然后根據(jù)驗證集的表現(xiàn)如mAP曲線、損失曲線進行針對性調(diào)整。5. 模型訓(xùn)練、評估與常見問題排錯5.1 啟動訓(xùn)練與監(jiān)控指標(biāo)使用一行命令即可開始訓(xùn)練yolo taskdetect modetrain modelyolov8s.pt datadataset.yaml epochs150 imgsz640 batch16 workers8訓(xùn)練開始后Ultralytics會啟動一個本地Web服務(wù)器通常是在http://localhost:3000提供實時監(jiān)控儀表盤。這是觀察訓(xùn)練進程的窗口。需要重點關(guān)注的指標(biāo)有損失曲線Loss Curvestrain/box_loss,train/cls_loss,train/dfl_loss訓(xùn)練集上的各項損失。理想情況下它們應(yīng)該隨著訓(xùn)練輪數(shù)平穩(wěn)下降。val/box_loss,val/cls_loss,val/dfl_loss驗證集上的損失。這是判斷模型是否過擬合的關(guān)鍵。如果訓(xùn)練損失持續(xù)下降但驗證損失在某個點后開始上升或停滯很可能發(fā)生了過擬合。性能指標(biāo)Performance Metricsmetrics/mAP50(B)在IoU閾值為0.5時的平均精度均值。這是最核心的指標(biāo)直接反映了模型檢測的準(zhǔn)確度。它會逐步上升并趨于平穩(wěn)。metrics/mAP50-95(B)在IoU閾值從0.5到0.95步長0.05區(qū)間內(nèi)的平均mAP。這是一個更嚴(yán)格的指標(biāo)要求邊界框定位更精確。metrics/precision(B),metrics/recall(B)精確率和召回率。我們希望兩者都高。高精度低召回說明模型保守漏檢多低精度高召回說明模型激進誤檢多。如何判斷訓(xùn)練是否良好訓(xùn)練/驗證損失平穩(wěn)下降后趨于收斂。mAP50和mAP50-95持續(xù)上升后在一個高點穩(wěn)定。驗證集損失沒有明顯高于訓(xùn)練集損失且沒有在后期反彈。如果訓(xùn)練過程不理想就需要進行問題排查。5.2 常見問題、原因分析與解決方案以下是我在訓(xùn)練吸煙檢測模型過程中遇到的一些典型問題及解決方法問題現(xiàn)象可能原因解決方案與排查步驟訓(xùn)練損失居高不下或波動劇烈1. 學(xué)習(xí)率(lr0)設(shè)置過高。2. 數(shù)據(jù)標(biāo)注存在大量錯誤或噪聲。3. 數(shù)據(jù)預(yù)處理或增強過于激進導(dǎo)致圖像難以學(xué)習(xí)。1.降低學(xué)習(xí)率嘗試lr00.01或0.001。2.檢查標(biāo)注質(zhì)量從訓(xùn)練集中隨機抽樣幾十張圖片用訓(xùn)練中的模型進行預(yù)測直觀查看模型學(xué)到的和標(biāo)注的差異。錯誤標(biāo)注會嚴(yán)重干擾學(xué)習(xí)。3.簡化數(shù)據(jù)增強暫時關(guān)閉Mosaic、MixUp等復(fù)雜增強只保留基礎(chǔ)的色彩抖動看損失是否穩(wěn)定。驗證損失遠高于訓(xùn)練損失且差距持續(xù)擴大過擬合1. 訓(xùn)練數(shù)據(jù)量太少。2. 模型復(fù)雜度如用了yolov8l或x相對于數(shù)據(jù)量過高。3. 數(shù)據(jù)增強不夠模型記住了訓(xùn)練集的特有噪聲。1.增加數(shù)據(jù)收集更多數(shù)據(jù)或使用更強大的數(shù)據(jù)增強如Mosaic, MixUp, CutMix。2.使用更小的模型從yolov8n或s開始。3.增加正則化適當(dāng)增大weight_decay參數(shù)如從0.0005調(diào)到0.001。4.早停Early Stopping監(jiān)控驗證損失在其連續(xù)多個epoch不下降時停止訓(xùn)練。mAP50很低但召回率(Recall)很高模型產(chǎn)生了大量誤檢False Positives。把很多不是香煙的物體如筆、手指也檢測出來了。1.檢查負樣本訓(xùn)練集中是否包含了足夠多的“不含香煙”的圖片需要確保數(shù)據(jù)集中有相當(dāng)比例的負樣本non_smoking標(biāo)簽的圖片讓模型學(xué)習(xí)什么是“不是香煙”。2.清理相似物在數(shù)據(jù)收集中是否有大量與香煙形狀顏色相似的物體可以考慮在數(shù)據(jù)集中加入一些“困難負樣本”如筆、筷子等并明確標(biāo)注為背景。3.提高分類損失權(quán)重(cls_pw)讓模型更關(guān)注分類的正確性。mAP50尚可但召回率很低模型漏檢很多False Negatives。很多香煙沒有被檢測出來。1.關(guān)注小目標(biāo)香煙是小目標(biāo)。檢查數(shù)據(jù)增強是否過度如隨機縮放后目標(biāo)變得更小?？梢詼p小imgsz如從640降到320不對于小目標(biāo)增大imgsz如到832可能更有效因為更大的輸入分辨率保留了更多細節(jié)。同時確保Mosaic增強是開啟的這對小目標(biāo)檢測有益。2.檢查標(biāo)注完整性是否存在大量未標(biāo)注的香煙實例尤其是遠處、模糊的小香煙。3.調(diào)整模型neck/head對于YOLOv8可以嘗試使用帶P2更高分辨率特征層的小目標(biāo)檢測專用模型如果官方有提供或者在自定義模型時關(guān)注淺層特征。訓(xùn)練后期性能突然崩潰指標(biāo)驟降1. 學(xué)習(xí)率調(diào)度問題。2. 訓(xùn)練數(shù)據(jù)中存在某些損壞的圖片或標(biāo)注文件。3. 硬件或軟件偶然錯誤。1.檢查學(xué)習(xí)率曲線是否在后期學(xué)習(xí)率降得太低或有個異常跳變可以嘗試使用余弦退火等更平滑的調(diào)度器。2.檢查數(shù)據(jù)完整性運行一個腳本遍歷所有圖片和標(biāo)簽文件確保都能正常打開和讀取。3.從崩潰前的檢查點恢復(fù)訓(xùn)練YOLOv8會定期保存檢查點.pt文件可以加載崩潰前最后一個好的檢查點降低學(xué)習(xí)率繼續(xù)訓(xùn)練。5.3 模型導(dǎo)出與部署測試訓(xùn)練完成后模型會保存在runs/detect/train/weights/目錄下其中best.pt是驗證集上表現(xiàn)最好的權(quán)重。我們可以用這個模型在測試集上看看最終效果yolo taskdetect modeval modelruns/detect/train/weights/best.pt datadataset.yaml然后用一些從未見過的圖片或視頻進行推理測試這是真正的“實戰(zhàn)”# 圖片推理 yolo taskdetect modepredict modelruns/detect/train/weights/best.pt sourcepath/to/test_image.jpg # 視頻推理 yolo taskdetect modepredict modelruns/detect/train/weights/best.pt sourcepath/to/test_video.mp4部署前的最后檢查速度測試在不同的硬件CPU, 邊緣計算設(shè)備上測試推理速度FPS看是否滿足應(yīng)用要求。場景泛化性找一些與訓(xùn)練集分布不同的圖片如不同國家的香煙包裝、電子煙、雪茄等測試模型的魯棒性。如果泛化性差說明數(shù)據(jù)多樣性仍需加強。錯誤案例分析仔細查看模型在測試集上預(yù)測錯誤的案例漏檢、誤檢分析原因。是標(biāo)注問題是數(shù)據(jù)缺失如某種特定角度還是模型能力上限這些分析是迭代優(yōu)化數(shù)據(jù)集和模型的最寶貴輸入。構(gòu)建一個高質(zhì)量的“吸煙數(shù)據(jù)集”遠不止是收集和打標(biāo)簽它是一個系統(tǒng)工程涵蓋了需求分析、數(shù)據(jù)采集、標(biāo)準(zhǔn)制定、工具流搭建、質(zhì)量控制、模型訓(xùn)練調(diào)優(yōu)和問題診斷的全流程。其中清晰的標(biāo)注規(guī)范和持續(xù)的質(zhì)量控制是保證數(shù)據(jù)質(zhì)量的靈魂而基于數(shù)據(jù)洞察的模型調(diào)參則是提升性能的關(guān)鍵。這個過程充滿了反復(fù)和調(diào)試但當(dāng)看到自己親手打造的數(shù)據(jù)集訓(xùn)練出的模型能準(zhǔn)確地識別出各種場景下的吸煙行為時那種成就感是完全不一樣的。希望這份詳細的記錄能為你構(gòu)建自己的定制化數(shù)據(jù)集提供一個扎實的路線圖。本文還有配套的精品資源點擊獲取