簡介這份鋼鐵表面缺陷檢測圖像數(shù)據(jù)集面向從事機器視覺與工業(yè)質(zhì)檢研究的開發(fā)者、學(xué)生及算法工程師用于訓(xùn)練和評估裂紋、銹蝕、凹坑、劃痕等表面缺陷的識別模型可作為課程實驗、畢業(yè)設(shè)計或缺陷檢測算法對比的樣本來源。壓縮包為zip格式共約2000個文件主體是1800張jpg缺陷圖像并配有1400余個xml標(biāo)注文件與1400余個txt說明文件另有data、names等配置項整體約26.81MB體積輕便易于本地加載。數(shù)據(jù)集已按訓(xùn)練、驗證、測試等劃分整理每張圖像均帶類別標(biāo)簽可直接用于監(jiān)督學(xué)習(xí)下的模型訓(xùn)練與精度、召回率、F1分數(shù)等指標(biāo)驗證省去自行標(biāo)注與切分的成本。目前已有2453人學(xué)習(xí)下載適合希望快速搭建缺陷檢測基線并復(fù)現(xiàn)實驗的讀者參考使用。1. 鋼鐵表面缺陷檢測圖像數(shù)據(jù)集1800 張帶標(biāo)簽圖像到底能撐起什么產(chǎn)線上跑視覺檢測的同行大概都遇到過這種局面算法框架搭好了模型結(jié)構(gòu)也調(diào)通了卡住的卻是數(shù)據(jù)——要么樣本量不夠要么缺陷類別標(biāo)注得含糊要么圖像質(zhì)量跟現(xiàn)場相機對不上。鋼鐵表面缺陷檢測圖像數(shù)據(jù)集1800 張圖像含標(biāo)簽這類資源解決的正是這個冷啟動問題。它面向的是熱軋、冷軋帶鋼表面常見的夾雜、劃痕、麻點、氧化皮壓入等缺陷類型圖像帶分類或檢測標(biāo)簽適合做基線模型訓(xùn)練、數(shù)據(jù)增強策略驗證、以及小樣本條件下的遷移學(xué)習(xí)實驗。如果你正在做工業(yè)質(zhì)檢方向的算法預(yù)研或者要給一條新產(chǎn)線快速搭一個能跑起來的缺陷識別原型這個量級的數(shù)據(jù)集是夠用的起點但如果你指望直接拿它上線那得先看清它的邊界在哪。2. 1800 張鋼鐵缺陷圖像的數(shù)據(jù)結(jié)構(gòu)與標(biāo)簽體系怎么讀拿到一個圖像數(shù)據(jù)集第一件事不是急著喂模型而是把目錄結(jié)構(gòu)、標(biāo)簽格式、類別分布摸清楚。1800 張這個量級說大不大說小也不小——如果類別均衡、標(biāo)注質(zhì)量過關(guān)訓(xùn)練一個中等規(guī)模的分類網(wǎng)絡(luò)綽綽有余但如果長尾嚴重少數(shù)類可能只有幾十張直接訓(xùn)練必然翻車。所以這一章先把「怎么讀數(shù)據(jù)」講透再談怎么用。2.1 目錄結(jié)構(gòu)與標(biāo)簽文件的三種常見組織方式鋼鐵表面缺陷數(shù)據(jù)集常見的組織方式有三種按類別分文件夾class_name/*.jpg、圖像與標(biāo)注文件分離images/labels/、以及單文件 CSV/JSON 索引。前兩種在分類和檢測任務(wù)里最常見第三種多見于從標(biāo)注平臺導(dǎo)出的原始數(shù)據(jù)。不管你拿到的是哪種第一步都是寫個腳本把結(jié)構(gòu)打印出來別靠肉眼翻文件夾。import os from collections import Counter def inspect_dataset(root): 打印數(shù)據(jù)集目錄結(jié)構(gòu)與文件類型分布 ext_counter Counter() class_counter Counter() for dirpath, dirnames, filenames in os.walk(root): for f in filenames: ext os.path.splitext(f)[1].lower() ext_counter[ext] 1 # 假設(shè)按類別分文件夾取父目錄名作為類別 parent os.path.basename(dirpath) if ext in (.jpg, .jpeg, .png, .bmp): class_counter[parent] 1 print(文件擴展名分布:, dict(ext_counter)) print(各類別圖像數(shù)量:, dict(class_counter)) return class_counter # 用法把 root 換成你的數(shù)據(jù)集根目錄 # inspect_dataset(./steel_defect_dataset)這段腳本做兩件事統(tǒng)計文件擴展名確認圖像格式是否統(tǒng)一鋼鐵數(shù)據(jù)集里常見 jpg 和 bmp 混用bmp 無損但體積大jpg 有壓縮偽影混用會影響模型對紋理的敏感度統(tǒng)計每個類別目錄下的圖像數(shù)量快速判斷類別均衡性。參數(shù)上沒什么可調(diào)的root指向數(shù)據(jù)集根目錄即可。跑完之后你會得到一張類別分布表如果發(fā)現(xiàn)某個類別少于 100 張后面訓(xùn)練時就得考慮過采樣或類別加權(quán)。2.2 類別分布統(tǒng)計與長尾問題的量化判斷1800 張圖像如果分 6 類平均每類 300 張聽起來還行但實際工業(yè)數(shù)據(jù)集往往不是均勻分布——正常樣本可能占一半某些罕見缺陷只有幾十張。長尾問題不量化就沒法決策所以得算兩個指標(biāo)最大類與最小類的比值不平衡比以及每類樣本數(shù)是否低于某個閾值。指標(biāo)計算方式判斷標(biāo)準(zhǔn)處理建議不平衡比最大類數(shù)量 / 最小類數(shù)量 10 視為嚴重長尾過采樣 類別加權(quán)損失最小類樣本數(shù)直接統(tǒng)計 100 張需警惕數(shù)據(jù)增強或遷移學(xué)習(xí)類別數(shù)標(biāo)簽去重610 類較常見過多則考慮合并細類我一般會先跑一遍統(tǒng)計把不平衡比算出來。如果比值在 3 以內(nèi)直接訓(xùn)練問題不大3 到 10 之間用加權(quán)交叉熵就能壓住超過 10光靠損失函數(shù)不夠得在數(shù)據(jù)層面做文章——對少數(shù)類做旋轉(zhuǎn)、翻轉(zhuǎn)、亮度擾動或者用切圖crop把大圖里的缺陷區(qū)域裁出來擴充樣本。鋼鐵缺陷有個特點劃痕和夾雜在紋理上差異明顯但麻點和氧化皮壓入在低分辨率下容易混淆所以增強時別過度模糊否則類間邊界更糊。2.3 標(biāo)簽格式轉(zhuǎn)換從分類目錄到檢測標(biāo)注的落地路徑如果標(biāo)簽是按類別分文件夾給的那它天然適合分類任務(wù)但很多鋼鐵缺陷檢測場景需要的是邊界框這時候就得看數(shù)據(jù)集是否附帶檢測標(biāo)注如 VOC XML 或 YOLO txt。如果只有分類標(biāo)簽?zāi)阌袃蓷l路一是先做分類基線驗證特征可分性二是用弱監(jiān)督或人工補標(biāo)的方式生成框標(biāo)注。常見做法是先用分類模型跑一遍看混淆矩陣再決定要不要投入標(biāo)注成本。import xml.etree.ElementTree as ET import os def voc_to_yolo(xml_path, img_w, img_h, class_map): 將 VOC 格式 XML 轉(zhuǎn)為 YOLO txt 格式 tree ET.parse(xml_path) root tree.getroot() lines [] for obj in root.findall(object): cls_name obj.find(name).text if cls_name not in class_map: continue cls_id class_map[cls_name] bbox obj.find(bndbox) xmin float(bbox.find(xmin).text) ymin float(bbox.find(ymin).text) xmax float(bbox.find(xmax).text) ymax float(bbox.find(ymax).text) # YOLO 格式中心點坐標(biāo) 寬高均歸一化到 0~1 x_center (xmin xmax) / 2.0 / img_w y_center (ymin ymax) / 2.0 / img_h w (xmax - xmin) / img_w h (ymax - ymin) / img_h lines.append(f{cls_id} {x_center:.6f} {y_center:.6f} {w:.6f} {h:.6f}) return lines # class_map 示例{scratch: 0, inclusion: 1, pitted: 2}轉(zhuǎn)換邏輯不復(fù)雜但有兩個坑一是圖像尺寸必須和 XML 里記錄的一致如果數(shù)據(jù)集做過縮放而 XML 沒同步更新歸一化坐標(biāo)會全錯二是類別名到 ID 的映射必須固定別在不同腳本里各寫一套否則訓(xùn)練時標(biāo)簽錯位模型學(xué)出來的東西完全是玄學(xué)。參數(shù)上img_w和img_h建議直接從圖像文件讀取別硬編碼。3. 用這 1800 張圖訓(xùn)練缺陷分類基線從切分到首輪評估數(shù)據(jù)摸清楚了接下來就是跑一個能出數(shù)的基線。1800 張圖不算多別一上來就上大模型先用輕量網(wǎng)絡(luò)把流程跑通拿到混淆矩陣和每類 F1再決定往哪個方向加投入。這一章按「切分 → 訓(xùn)練 → 評估 → 看錯例」的順序走每一步都給可復(fù)現(xiàn)的配置。3.1 訓(xùn)練集/驗證集/測試集切分的分層策略圖像分類任務(wù)最常見的翻車點之一是切分不分層——隨機切完之后驗證集里某個類別一張都沒有評估結(jié)果完全不可信。鋼鐵缺陷數(shù)據(jù)尤其要注意如果按文件夾隨機抽少數(shù)類很容易在驗證集里消失。正確做法是按類別分層抽樣保證每個集合里各類別比例接近原始分布。import os import random import shutil from collections import defaultdict def stratified_split(root, out_root, ratios(0.7, 0.15, 0.15), seed42): 按類別分層切分數(shù)據(jù)集為 train/val/test random.seed(seed) class_files defaultdict(list) for cls in os.listdir(root): cls_dir os.path.join(root, cls) if not os.path.isdir(cls_dir): continue for f in os.listdir(cls_dir): if f.lower().endswith((.jpg, .png, .bmp)): class_files[cls].append(os.path.join(cls_dir, f)) for cls, files in class_files.items(): random.shuffle(files) n len(files) n_train int(n * ratios[0]) n_val int(n * ratios[1]) splits { train: files[:n_train], val: files[n_train:n_train n_val], test: files[n_train n_val:] } for split, split_files in splits.items(): dst os.path.join(out_root, split, cls) os.makedirs(dst, exist_okTrue) for f in split_files: shutil.copy(f, dst) # stratified_split(./steel_defect_dataset, ./split_dataset)這段腳本的關(guān)鍵在defaultdict那一步先按類別把文件歸組再在組內(nèi)打亂切分保證每個集合都覆蓋所有類別。ratios默認 7:1.5:1.51800 張圖對應(yīng)訓(xùn)練集約 1260 張、驗證和測試各約 270 張。如果某個類別總數(shù)不到 20 張切完之后測試集可能只剩兩三張這時候建議把比例調(diào)成 8:1:1或者對少數(shù)類做留一法評估。seed固定住保證每次切分結(jié)果一致方便復(fù)現(xiàn)。3.2 輕量卷積網(wǎng)絡(luò)的首輪訓(xùn)練配置與關(guān)鍵超參基線模型我一般選 ResNet18 或 MobileNetV3-Small輸入尺寸 224×224batch size 32初始學(xué)習(xí)率 1e-3優(yōu)化器用 AdamW訓(xùn)練 30 個 epoch 起步。鋼鐵缺陷的紋理特征比較細輸入尺寸別壓得太狠160×160 以下劃痕類缺陷容易丟細節(jié)。數(shù)據(jù)增強用隨機水平翻轉(zhuǎn)、±15 度旋轉(zhuǎn)、亮度對比度擾動別用太激進的裁剪否則缺陷區(qū)域可能被裁掉。import torch import torch.nn as nn from torchvision import models, transforms # 數(shù)據(jù)增強與歸一化 train_tf transforms.Compose([ transforms.Resize((224, 224)), transforms.RandomHorizontalFlip(), transforms.RandomRotation(15), transforms.ColorJitter(brightness0.2, contrast0.2), transforms.ToTensor(), transforms.Normalize(mean[0.485, 0.456, 0.406], std[0.229, 0.224, 0.225]) ]) # 加載預(yù)訓(xùn)練 ResNet18替換分類頭 model models.resnet18(weightsIMAGENET1K_V1) num_classes 6 # 按實際類別數(shù)修改 model.fc nn.Linear(model.fc.in_features, num_classes) # 類別加權(quán)損失緩解長尾 class_weights torch.tensor([1.0, 1.5, 2.0, 1.2, 3.0, 1.8]) # 按類別頻率反比設(shè)置 criterion nn.CrossEntropyLoss(weightclass_weights) optimizer torch.optim.AdamW(model.parameters(), lr1e-3, weight_decay1e-4)weightsIMAGENET1K_V1表示用 ImageNet 預(yù)訓(xùn)練權(quán)重初始化這在 1800 張的小數(shù)據(jù)集上是關(guān)鍵——從零訓(xùn)練幾乎必然過擬合。class_weights那行要按你實際的類別分布來調(diào)頻率越低的類權(quán)重越高但別超過 5否則少數(shù)類的梯度會主導(dǎo)更新把多數(shù)類的性能拉垮。學(xué)習(xí)率 1e-3 配合 AdamW 是比較穩(wěn)的起點如果 loss 在前幾個 epoch 震蕩厲害降到 5e-4。3.3 混淆矩陣與每類 F1 的讀法判斷數(shù)據(jù)集能不能用訓(xùn)練跑完別只看總體準(zhǔn)確率。1800 張圖里如果正常樣本占一半模型全猜正常也能有 50% 準(zhǔn)確率這個數(shù)沒有意義。真正要看的是混淆矩陣和每類 F1。鋼鐵缺陷里最容易混的是麻點和氧化皮壓入——兩者在灰度圖上都表現(xiàn)為暗斑區(qū)別在邊緣銳度和分布密度。如果這兩類的混淆矩陣對角線很淡說明要么標(biāo)注邊界不清要么分辨率不夠這時候加數(shù)據(jù)不如先回頭查標(biāo)注一致性。評估指標(biāo)關(guān)注點異常信號每類召回率少數(shù)類是否被漏檢某類召回 0.5 需查樣本量每類精確率誤檢集中在哪兩類之間兩類互相混淆說明特征重疊宏平均 F1類別不均衡下的整體表現(xiàn)與準(zhǔn)確率差距大說明長尾嚴重我一般會把混淆矩陣畫出來盯著非對角線上的大數(shù)看。如果發(fā)現(xiàn)某兩類互相錯得厲害先別急著調(diào)模型把錯例圖調(diào)出來看幾張——很多時候是標(biāo)注本身就有歧義模型只是忠實地學(xué)了噪聲。4. 鋼鐵缺陷檢測的避坑清單五條血淚經(jīng)驗這一章單獨拎出來講踩坑因為鋼鐵表面缺陷這個場景有幾個特有的陷阱不提前說清楚后面調(diào)參調(diào)到頭禿也找不到原因。下面五條都是實際做下來反復(fù)遇到的按「現(xiàn)象 → 原因 → 解決」寫。4.1 圖像分辨率與現(xiàn)場相機不匹配導(dǎo)致指標(biāo)虛高現(xiàn)象離線訓(xùn)練集上 F1 跑到 0.92換到現(xiàn)場相機采的圖直接掉到 0.6。原因數(shù)據(jù)集里的圖像可能是經(jīng)過裁剪或縮放的分辨率和現(xiàn)場采集的不一致模型學(xué)到的紋理尺度對不上。解決訓(xùn)練前統(tǒng)一確認圖像的實際物理分辨率每毫米像素數(shù)如果現(xiàn)場相機參數(shù)不同在預(yù)處理里加一步重采樣把尺度對齊另外在驗證集里混入少量現(xiàn)場圖做域適應(yīng)評估別只看公開集的數(shù)。4.2 標(biāo)簽噪聲同一類缺陷在不同圖像里標(biāo)注口徑不一致現(xiàn)象模型對某一類的預(yù)測忽好忽壞混淆矩陣里該類分散到多個類。原因標(biāo)注時對類別邊界定義不統(tǒng)一比如有的標(biāo)注者把淺劃痕歸到劃痕有的歸到麻點。解決抽 50 張圖人工復(fù)核統(tǒng)計類間歧義比例如果超過 10%要么合并細類要么重新制定標(biāo)注規(guī)范再補標(biāo)。鋼鐵缺陷的類別定義最好帶圖示說明光靠文字描述不同人理解差異很大。4.3 過度依賴亮度增強把缺陷特征抹掉現(xiàn)象加了 ColorJitter 之后驗證集指標(biāo)反而降了。原因鋼鐵缺陷很多是靠灰度對比度區(qū)分的亮度擾動過大把弱缺陷的信號壓沒了。解決亮度擾動幅度控制在 ±0.1 以內(nèi)對比度擾動可以稍大但別和亮度同時拉滿更好的做法是用直方圖均衡化做預(yù)處理而不是在增強里隨機改亮度。4.4 小樣本類別直接過采樣導(dǎo)致過擬合現(xiàn)象少數(shù)類召回上去了但精確率崩了誤檢大量增加。原因簡單復(fù)制少數(shù)類樣本模型記住了具體圖像而不是特征。解決用過采樣時配合強增強旋轉(zhuǎn)、彈性變形或者改用 focal loss 讓模型聚焦難樣本更穩(wěn)的做法是先做遷移學(xué)習(xí)用預(yù)訓(xùn)練特征凍結(jié)前幾層只微調(diào)分類頭。4.5 訓(xùn)練/測試切分時同一張原圖的增強版本跨集泄漏現(xiàn)象測試集指標(biāo)好得不真實換一批新圖就露餡。原因先做數(shù)據(jù)增強再切分同一張原圖的不同增強版本同時出現(xiàn)在訓(xùn)練集和測試集。解決切分必須在增強之前做按原圖 ID 分組切分保證同一原圖的所有變體只落在一個集合里。這個坑在圖像分類里特別隱蔽因為增強后的圖看起來不一樣但模型其實見過原圖。5. 從 1800 張到產(chǎn)線可用小樣本下的遷移與驗證技巧1800 張圖做原型夠用但離產(chǎn)線還有距離。這一章講兩個進階方向怎么用遷移學(xué)習(xí)把有限樣本的效用榨干以及怎么設(shè)計一套能說服自己的驗證流程。最后給一個我常用的技巧——用「留出產(chǎn)線批次」做最終檢驗而不是只看隨機切分的測試集。5.1 凍結(jié)策略與分層學(xué)習(xí)率的實操配置小數(shù)據(jù)集上微調(diào)預(yù)訓(xùn)練模型凍結(jié)多少層很關(guān)鍵。全凍結(jié)只訓(xùn)分類頭收斂快但特征適配不足全解凍又容易過擬合。我一般分三段前兩個 stage 凍結(jié)第三個 stage 和分類頭用較小學(xué)習(xí)率1e-4分類頭本身用 1e-3。這樣底層通用特征不動中層紋理特征小幅調(diào)整分類頭快速適配新類別。# 分層學(xué)習(xí)率凍結(jié)前兩層其余層用小學(xué)習(xí)率 for name, param in model.named_parameters(): if layer1 in name or layer2 in name: param.requires_grad False optimizer torch.optim.AdamW([ {params: [p for n, p in model.named_parameters() if fc in n], lr: 1e-3}, {params: [p for n, p in model.named_parameters() if fc not in n and p.requires_grad], lr: 1e-4} ], weight_decay1e-4)這段配置的核心是把分類頭和骨干網(wǎng)絡(luò)分開設(shè)學(xué)習(xí)率。requires_grad False凍掉前兩層減少可訓(xùn)練參數(shù)1800 張圖不至于把整個網(wǎng)絡(luò)帶偏。如果驗證集指標(biāo)在 10 個 epoch 內(nèi)不漲可以把 layer3 也解凍試試但學(xué)習(xí)率再降一檔。5.2 用留出批次驗證代替隨機切分更接近產(chǎn)線的評估方式隨機切分的測試集有個問題同一批生產(chǎn)條件下拍的圖外觀高度相似隨機切分會讓訓(xùn)練集和測試集共享大量相似紋理指標(biāo)偏樂觀。更接近真實的做法是按采集批次留出——比如按時間段或鋼卷編號分組留出整批做測試。這樣能暴露模型對光照變化、批次間表面狀態(tài)差異的敏感度。具體操作是在切分腳本里加一個group_key參數(shù)按組切而不是按文件切。驗證方式優(yōu)點局限適用階段隨機分層切分實現(xiàn)簡單類別均衡指標(biāo)偏樂觀基線快速驗證按批次留出接近產(chǎn)線分布偏移需要批次元信息上線前評估跨域測試檢驗泛化邊界需額外數(shù)據(jù)源模型選型對比5.3 一個習(xí)慣先看錯例再調(diào)參最后說個我自己的習(xí)慣。每次指標(biāo)不達預(yù)期我第一件事不是改學(xué)習(xí)率或換模型而是把錯例圖導(dǎo)出來看二十張。鋼鐵缺陷的錯例往往能直接告訴你問題在哪——如果錯例全是低對比度的麻點那是預(yù)處理對比度不夠如果錯例集中在圖像邊緣那是切圖時缺陷被截斷如果錯例的標(biāo)注框明顯偏了那是標(biāo)注質(zhì)量問題調(diào)模型純屬白費功夫。這個習(xí)慣幫我省下的時間比任何調(diào)參技巧都多。數(shù)據(jù)量不大的時候人對錯例的直覺比指標(biāo)更可靠。希望幫到你。本文還有配套的精品資源點擊獲取