簡(jiǎn)介這份資源是面向 AnyLabeling 標(biāo)注工具用戶的 Segment AnythingViT-B模型權(quán)重包主要解決在本地自動(dòng)標(biāo)注場(chǎng)景中缺少可用分割模型的問(wèn)題適合已具備一定深度學(xué)習(xí)環(huán)境配置經(jīng)驗(yàn)、希望用 SAM 提升標(biāo)注效率的開(kāi)發(fā)者與算法學(xué)習(xí)者。壓縮包共 3 個(gè)文件包含 2 個(gè) onnx 模型文件與 1 個(gè) yaml 配置文件分別承擔(dān)編碼器、解碼器推理與參數(shù)配置職責(zé)整體約 332.26MB解壓后放入指定模型目錄即可被 AnyLabeling 識(shí)別調(diào)用。目前已有 1255 人學(xué)習(xí)下載說(shuō)明該模型在自動(dòng)標(biāo)注工作流中具有較高實(shí)用價(jià)值。借助這份權(quán)重讀者可快速搭建 SAM 自動(dòng)分割能力用于圖像掩碼生成、半自動(dòng)標(biāo)注與數(shù)據(jù)預(yù)處理減少手工勾畫成本并為后續(xù)微調(diào)或集成到自有標(biāo)注流程提供可直接運(yùn)行的模型基礎(chǔ)。1. 拆開(kāi) sam-vit-b-01ec64.zipAnyLabeling 里那個(gè) ViT-B 權(quán)重到底怎么用如果你正在用 AnyLabeling 做標(biāo)注多半在模型列表里見(jiàn)過(guò)Segment Anything (ViT-B)這個(gè)選項(xiàng)點(diǎn)下去之后軟件會(huì)提示你下載一個(gè)叫sam-vit-b-01ec64.zip的包。很多人第一次遇到會(huì)愣一下這玩意兒是模型本體還是配置文件下完放哪為什么我下完還是加載失敗我當(dāng)初也是這么過(guò)來(lái)的翻了一圈文檔沒(méi)找到完整說(shuō)明最后自己把包拆開(kāi)、對(duì)著日志一點(diǎn)點(diǎn)試出來(lái)的。這個(gè) zip 里裝的是 Meta Segment Anything 的 ViT-B 版本權(quán)重配合 AnyLabeling 的自動(dòng)分割功能使用能讓你在標(biāo)注時(shí)點(diǎn)一下就把目標(biāo)輪廓摳出來(lái)省掉大量手動(dòng)畫多邊形的時(shí)間。它適合做圖像標(biāo)注、數(shù)據(jù)清洗、半自動(dòng)標(biāo)注流水線的從業(yè)者尤其是手頭有幾千張圖要標(biāo)、又不想純手工硬啃的人。下面我把這個(gè)包的來(lái)龍去脈、放置路徑、參數(shù)含義和幾個(gè)我踩過(guò)的坑一次講清楚。2. 先搞懂 SAM 與 ViT-B為什么 AnyLabeling 偏偏選這個(gè)組合2.1 Segment Anything 的「提示式分割」到底在做什么Segment Anything ModelSAM和傳統(tǒng)分割模型最大的區(qū)別是它不依賴固定類別。你給它一張圖再給一個(gè)提示——一個(gè)點(diǎn)、一個(gè)框、或者一段粗略的掩碼——它就能返回對(duì)應(yīng)的分割結(jié)果。這個(gè)能力叫 promptable segmentation本質(zhì)是把分割任務(wù)從「訓(xùn)練時(shí)定好類別」變成「推理時(shí)由提示驅(qū)動(dòng)」。AnyLabeling 正是利用這一點(diǎn)你在圖上點(diǎn)一下目標(biāo)它把點(diǎn)坐標(biāo)作為提示喂給 SAMSAM 返回掩碼AnyLabeling 再把掩碼轉(zhuǎn)成多邊形標(biāo)注。整個(gè)過(guò)程不需要你預(yù)先定義「這是貓還是狗」對(duì)標(biāo)注場(chǎng)景非常友好因?yàn)闃?biāo)注時(shí)你往往只關(guān)心「把這塊摳出來(lái)」不關(guān)心類別名。SAM 的架構(gòu)分三塊圖像編碼器Image Encoder、提示編碼器Prompt Encoder、掩碼解碼器Mask Decoder。圖像編碼器是重頭戲它把輸入圖像編碼成特征這部分計(jì)算量大但每張圖只跑一次提示編碼器和掩碼解碼器很輕可以反復(fù)快速出結(jié)果。這也是為什么 SAM 能在交互式標(biāo)注里做到「點(diǎn)一下幾乎立刻出輪廓」。2.2 ViT-B 在 SAM 家族里的定位與選型理由SAM 官方放出過(guò)三個(gè)規(guī)模的圖像編碼器ViT-B、ViT-L、ViT-H。B 是 BaseL 是 LargeH 是 Huge。參數(shù)量、顯存占用、推理速度依次遞增分割精度也依次遞增。ViT-B 是其中最小、最輕的一檔。AnyLabeling 默認(rèn)推薦 ViT-B原因很實(shí)際標(biāo)注是交互式操作你點(diǎn)一下要等結(jié)果如果每次等兩三秒標(biāo)注節(jié)奏就斷了。ViT-B 在普通消費(fèi)級(jí)顯卡甚至 CPU 上都能跑出可接受的響應(yīng)速度顯存占用也低8GB 顯存的機(jī)器基本無(wú)壓力。ViT-H 精度確實(shí)更好但那個(gè)顯存和延遲在標(biāo)注這種高頻交互場(chǎng)景里反而拖后腿。選型上我的建議是日常標(biāo)注、目標(biāo)邊緣不算特別復(fù)雜比如車輛、行人、家具、文檔區(qū)域ViT-B 完全夠用如果你標(biāo)的是醫(yī)學(xué)影像、遙感圖像這種邊緣極其精細(xì)、容錯(cuò)率低的任務(wù)再考慮上更大的模型但要有對(duì)應(yīng)的硬件。2.3 這個(gè) zip 里到底裝了什么sam-vit-b-01ec64.zip解壓后是一個(gè) PyTorch 權(quán)重文件命名類似sam_vit_b_01ec64.pth。01ec64是權(quán)重版本的哈希標(biāo)識(shí)用來(lái)區(qū)分不同訓(xùn)練輪次產(chǎn)出的權(quán)重。這個(gè)文件就是 ViT-B 圖像編碼器加提示編碼器、掩碼解碼器的全部參數(shù)體積在三百多 MB 量級(jí)。它不是配置文件也不是可執(zhí)行程序就是一個(gè)純權(quán)重。AnyLabeling 需要先有 SAM 的模型結(jié)構(gòu)代碼軟件內(nèi)置了再把這個(gè)權(quán)重加載進(jìn)去才能工作。所以「下載完放對(duì)位置」這一步很關(guān)鍵放錯(cuò)了軟件找不到就會(huì)一直提示你下載或加載失敗。提示不同來(lái)源的權(quán)重哈?？赡懿煌珹nyLabeling 認(rèn)的是它自己配置里指定的那個(gè)版本。別隨便拿一個(gè)同名文件替換哈希對(duì)不上可能加載報(bào)錯(cuò)。3. 把權(quán)重接進(jìn) AnyLabeling路徑、加載與首次推理驗(yàn)證3.1 權(quán)重放置路徑與目錄結(jié)構(gòu)AnyLabeling 對(duì)模型文件的查找有固定邏輯。它一般會(huì)在用戶目錄下建一個(gè)模型緩存文件夾把下載的權(quán)重放進(jìn)去。不同系統(tǒng)路徑不一樣常見(jiàn)的是WindowsC:\Users\用戶名\.anylabeling\models\Linux / macOS~/.anylabeling/models/你可以直接在 AnyLabeling 界面里點(diǎn)模型下載讓它自己下也可以手動(dòng)把sam-vit-b-01ec64.zip解壓后的.pth文件放進(jìn)這個(gè)目錄。手動(dòng)放的好處是網(wǎng)絡(luò)不穩(wěn)時(shí)不用反復(fù)重試壞處是路徑容易放錯(cuò)。放好之后目錄里應(yīng)該能看到權(quán)重文件本身而不是還套著一層 zip 或者多套了一層文件夾。我見(jiàn)過(guò)有人解壓出sam-vit-b-01ec64/sam_vit_b_01ec64.pth這種雙層結(jié)構(gòu)軟件在 models 目錄下直接找.pth找不到就報(bào)錯(cuò)。3.2 在界面里加載模型并跑通第一次分割路徑放對(duì)后打開(kāi) AnyLabeling在左側(cè)或頂部找到模型選擇入口選Segment Anything (ViT-B)。如果權(quán)重就位它會(huì)直接加載狀態(tài)欄或日志里會(huì)顯示模型已就緒。如果還在提示下載說(shuō)明它沒(méi)在預(yù)期路徑找到文件。加載成功后打開(kāi)一張圖用自動(dòng)標(biāo)注/智能分割工具在目標(biāo)上點(diǎn)一個(gè)點(diǎn)。正常情況下一兩秒內(nèi)會(huì)出現(xiàn)一個(gè)高亮掩碼覆蓋住目標(biāo)。你可以繼續(xù)加點(diǎn)修正或者切換成框選提示。這一步是驗(yàn)證權(quán)重是否真正可用的最快方式。如果點(diǎn)了沒(méi)反應(yīng)、報(bào)錯(cuò)、或者掩碼明顯錯(cuò)亂先別懷疑模型本身八成是路徑、版本或依賴問(wèn)題。3.3 用一段最小腳本獨(dú)立驗(yàn)證權(quán)重可用性有時(shí)候界面報(bào)錯(cuò)信息很含糊我習(xí)慣繞開(kāi) AnyLabeling直接用 Python 加載權(quán)重跑一次推理確認(rèn)文件本身沒(méi)壞。前提是你本地裝了 PyTorch 和 segment-anything 的代碼庫(kù)。import torch from segment_anything import sam_model_registry, SamPredictor import numpy as np from PIL import Image # 權(quán)重路徑按你實(shí)際放置位置改 ckpt_path sam_vit_b_01ec64.pth # 用 registry 按 vit_b 結(jié)構(gòu)加載權(quán)重 sam sam_model_registry[vit_b](checkpointckpt_path) # 有 GPU 就用 GPU沒(méi)有就 CPUViT-B 在 CPU 上也能跑 device cuda if torch.cuda.is_available() else cpu sam.to(devicedevice) predictor SamPredictor(sam) # 讀一張測(cè)試圖 image np.array(Image.open(test.jpg).convert(RGB)) predictor.set_image(image) # 給一個(gè)前景點(diǎn)提示坐標(biāo)是 (x, y) input_point np.array([[500, 375]]) input_label np.array([1]) # 1 表示前景點(diǎn) masks, scores, logits predictor.predict( point_coordsinput_point, point_labelsinput_label, multimask_outputTrue, ) print(masks shape:, masks.shape) print(scores:, scores)這段代碼的邏輯是先按vit_b結(jié)構(gòu)實(shí)例化模型并加載權(quán)重再把圖像送進(jìn)set_image做一次圖像編碼最后用點(diǎn)提示調(diào)predict拿掩碼。multimask_outputTrue會(huì)返回三個(gè)候選掩碼scores是它們的置信度通常取分?jǐn)?shù)最高的那個(gè)。參數(shù)上要注意input_label里 1 是前景點(diǎn)、0 是背景點(diǎn)點(diǎn)錯(cuò)標(biāo)簽結(jié)果會(huì)完全跑偏坐標(biāo)是圖像像素坐標(biāo)不是歸一化坐標(biāo)。如果這段腳本能跑出合理的masks shape一般是(3, H, W)說(shuō)明權(quán)重文件本身沒(méi)問(wèn)題界面里再出問(wèn)題就是 AnyLabeling 的配置或路徑問(wèn)題。4. 避坑與排查權(quán)重加載失敗、顯存爆掉、掩碼錯(cuò)亂怎么定位4.1 現(xiàn)象界面一直提示下載模型手動(dòng)放了也沒(méi)用原因通常有兩個(gè)一是路徑不對(duì)軟件找的目錄和你放的目錄不是同一個(gè)二是文件名或?qū)蛹?jí)不對(duì)多套了文件夾或者文件名被改過(guò)。解決先在軟件設(shè)置或日志里確認(rèn)它實(shí)際查找的模型目錄把.pth直接放到那一層。文件名保持原樣別自己重命名。放好后重啟軟件再試。4.2 現(xiàn)象加載時(shí)報(bào)哈希不匹配或版本錯(cuò)誤原因是你用的權(quán)重和 AnyLabeling 配置里期望的版本對(duì)不上。01ec64這個(gè)標(biāo)識(shí)就是用來(lái)做這個(gè)校驗(yàn)的換了別的哈希軟件可能拒絕加載。解決用軟件內(nèi)置下載拿到的那個(gè)包或者確認(rèn)你手上的包哈希與配置一致。別從不明來(lái)源隨便抓一個(gè)同名文件頂上。4.3 現(xiàn)象推理時(shí)顯存不足CUDA out of memory原因ViT-B 雖然輕但如果你同時(shí)開(kāi)了大圖、批量處理或者顯卡本身顯存很小還是會(huì)爆。另外有些人誤裝了 ViT-H 的權(quán)重卻以為是 B那顯存需求直接翻好幾倍。解決確認(rèn)加載的是 ViT-B 權(quán)重把輸入圖分辨率降下來(lái)關(guān)掉其他占顯存的程序?qū)嵲诓恍星?CPU 推理慢但能跑。4.4 現(xiàn)象點(diǎn)一下出的掩碼完全不對(duì)摳到別的地方原因多半是提示點(diǎn)坐標(biāo)或標(biāo)簽傳錯(cuò)了。比如把背景點(diǎn)標(biāo)成前景或者坐標(biāo)用了歸一化值而不是像素值。也可能是圖像預(yù)處理時(shí)被縮放但提示坐標(biāo)沒(méi)跟著換算。解決檢查坐標(biāo)是否在原圖尺度上確認(rèn)前景/背景標(biāo)簽如果 AnyLabeling 里圖被縮放過(guò)注意它的坐標(biāo)換算邏輯必要時(shí)先在原圖上驗(yàn)證。4.5 現(xiàn)象第一次推理特別慢后面才快起來(lái)原因這是正常的。圖像編碼器第一次跑要初始化、加載權(quán)重到顯存、編譯算子有冷啟動(dòng)開(kāi)銷。后面同一張圖或同尺寸圖會(huì)快很多。解決不用處理屬于預(yù)期行為。如果每次都慢檢查是不是每張圖都重新加載了模型那就要改成模型常駐。5. 進(jìn)階把 ViT-B 權(quán)重用進(jìn)批量半自動(dòng)標(biāo)注流水線單張交互標(biāo)注只是入門真正省時(shí)間的是把 SAM 接進(jìn)批量流程。思路是先用檢測(cè)模型或簡(jiǎn)單規(guī)則給出候選框再把框作為提示批量喂給 SAM讓它自動(dòng)出掩碼最后把掩碼轉(zhuǎn)成標(biāo)注格式落盤。這樣人只需要抽檢和修正不用每張圖都手點(diǎn)。具體做法上我一般會(huì)寫一個(gè)腳本循環(huán)讀圖對(duì)每張圖調(diào)用SamPredictor提示用框而不是點(diǎn)因?yàn)榭蛱崾緦?duì)批量場(chǎng)景更穩(wěn)定?？蛱崾镜挠梅ㄊ前裪nput_point換成input_box傳[x1, y1, x2, y2]。掩碼出來(lái)后用cv2.findContours或skimage.measure.find_contours提取輪廓再按 COCO 或 YOLO 的格式寫 json 或 txt。這里有個(gè)關(guān)鍵參數(shù)multimask_output在批量場(chǎng)景建議設(shè)False只出最可能的那一個(gè)省時(shí)間也省后處理邏輯。另外pred_iou_thresh和stability_score_thresh這類過(guò)濾閾值如果你用的是SamAutomaticMaskGenerator做全圖自動(dòng)分割可以調(diào)高一點(diǎn)減少碎掩碼但調(diào)太高會(huì)漏掉小目標(biāo)需要按你的數(shù)據(jù)試。驗(yàn)證批量結(jié)果是否靠譜我的習(xí)慣是隨機(jī)抽十張圖把生成的掩碼疊加回原圖看一眼重點(diǎn)看邊緣有沒(méi)有明顯溢出或缺失。如果邊緣普遍偏胖或偏瘦可能是提示框給得太松或太緊回去調(diào)檢測(cè)框的 padding。從那以后我每次拿到新的 SAM 權(quán)重都強(qiáng)制先跑一遍最小腳本驗(yàn)證再進(jìn)界面最后才上批量流程三步走完基本不會(huì)再被「玄學(xué)加載失敗」坑到。希望幫到你。本文還有配套的精品資源點(diǎn)擊獲取