鍚庡�勭悊缁撴��

segmentationmap=torchvision.transforms.functional.topilimage(output.squeeze())segmentationmap=cv2.cvtColor(np.array(segmentationmap),cv2.COLORRGB2BGR)

鍒涙剰鍗佽冻锛岃�掕壊绔嬩��

绾㈡�僣v鐨勷煍ユ极鐢诲垱鎰忓崄瓒筹紝姣忎竴涓�瑙掕壊閮芥湁鍏剁嫭鐗圭殑涓�鎬у拰鑳屾櫙鏁呬簨銆傛棤璁烘槸鍕囨暍鐨勮嫳闆勶紝杩樻槸绁炵�樼殑榄旀硶甯堬紝姣忎竴涓�瑙掕壊閮界粡杩囦簡娣卞叆鐨勮�捐�″拰濉戦�狅紝浣垮叾鍦ㄨ�昏�呭績涓�鐣欎笅娣卞埢鐨勷煍ュ嵃璞°�傝繖浜涜�掕壊鐨勬垚闀垮拰鍙樺寲锛屼篃鏄�婕�鐢绘晠浜嬬殑閲嶈�佺粍鎴愰儴鍒嗭紝璁╄�昏�呬滑鍦ㄩ槄璇昏繃绋嬩腑鎰熷彈鍒拌�掕壊鐨勭湡瀹炴儏鎰熷拰鍛借繍銆�

鎿嶄綔鎸囧崡锛�

鍑嗗�囬樁娈碉細鏍规嵁绀句氦濯掍綋鐨勭壒鐐癸紝閫夋嫨鍚堥�傜殑绾㈡�僣v妯℃澘銆傚彲浠ヤ娇鐢ㄧ畝娲併�佺編瑙傜殑璁捐�￠�庢牸锛岀獊鍑轰綘鐨勪釜浜虹壒鑹插拰鑱屼笟鑳屾櫙銆傝�捐�￠樁娈碉細浣跨敤绾㈡�僣v骞冲彴鎻愪緵鐨勫浘鐗囧拰鏂囧瓧绱犳潗锛屾牴鎹�鑷�宸辩殑涓�浜洪�庢牸璁捐�″嚭鐙�鐗圭殑cv銆傚彲浠ュ姞鍏ヤ竴浜涗釜浜虹収鐗囨垨鑰呮爣蹇楁�х殑鍥炬�堬紝澧炲姞涓�鎬у寲鐨勯瓍鍔涖��

灞曠ず闃舵�碉細鍦ㄧぞ浜ゅ獟浣撲笂鍒嗕韩浣犵殑绾㈡�僣v锛屽惛寮曞叧娉ㄨ�呯殑鍏磋叮锛屽苟馃摑涓庝粬浠�杩涜�屾繁鍏ヤ氦娴併�傞�氳繃绾㈡�僣v灞曠ず浣犵殑鑱屼笟鎴愬氨鍜屼釜浜哄搧鐗岋紝鍚稿紩鏇村�氱殑绮変笣鍜屽叧娉ㄣ��

鎿嶄綔鎸囧崡锛�

鍑嗗�囬樁娈碉細鏍规嵁浣犵殑鑱屼笟鐩�鏍囷紝閫夋嫨閫傚悎鐨勭孩妗僣v妯℃澘銆傚�傛灉浣犳�ｅ湪瀵绘壘鏂扮殑鑱屼笟鏈轰細锛屽彲浠ョ獊鍑轰綘鐨勬妧鑳藉拰鎴愬氨锛涘�傛灉浣犳�ｅ湪鍐呮帹鎴栬�呰法閮ㄩ棬璋冨姩锛屽彲浠ラ噸鐐瑰睍绀轰綘鐨勫洟闃熷悎浣滃拰棰嗗�艰兘鍔涖�傝�捐�￠樁娈碉細鏍规嵁浣犵殑鑱屼笟鍙戝睍璺�寰勶紝瀹氬埗绾㈡�僣v鐨勫唴瀹瑰拰璁捐�°�備娇鐢ㄤ竴浜涗笓涓氱殑鍥捐〃鍜屾暟鎹�锛屽睍绀轰綘鐨勬垚闀垮拰鑱屼笟鎴愬氨銆�

璋冩暣闃舵�碉細瀹氭湡鏇存柊绾㈡�僣v锛岀‘淇濆唴瀹圭殑鏃舵晥鎬у拰鍑嗙‘鎬с�傛牴鎹�鑱屼笟鍙戝睍鐨勯渶瑕侊紝璋冩暣閲嶇偣鍐呭�癸紝绐佸嚭褰撳墠鐨勮亴涓氫紭鍔裤�傚睍绀洪樁娈碉細鍦ㄨ亴涓氬彂灞曠殑鍚勪釜闃舵�碉紝閫氳繃绾㈡�僣v灞曠ず浣犵殑鎴愰暱杞ㄨ抗鍜屾湭鏉ョ殑鍙戝睍鏂瑰悜锛屽惛寮曟洿澶氱殑鍚堜綔鍜屾満浼氥��

鍚庡�勭悊缁撴��

segmentationmap=torchvision.transforms.functional.topilimage(output.squeeze())segmentationmap=cv2.cvtColor(np.array(segmentationmap),cv2.COLORRGB2BGR)

妯″瀷娴嬭瘯涓庨獙璇�

#鍔犺浇娴嬭瘯鏁版嵁test_loader=torch.utils.data.DataLoader(test_dataset,batch_size=32,shuffle=False)#娴嬭瘯妯″瀷model.eval()correct=0total=0withtorch.no_grad():forimages,labelsintest_loader:outputs=model(images)_,predicted=torch.max(outputs,1)total+=labels.size(0)correct+=(predicted==labels).sum().item()print(f'Accuracyofthemodelonthetestimages:{100*correct/total}%')

鏍″�癸細缃楀弸蹇�(1C0m4pJyqZtPma0S7t9ZFfz4hTykKag)