1. 主页 > 大智慧

工业质检场景实战:YOLOv5 v7上采样调优拯救电路板缺陷检测


馃敡鍦烘櫙鐥涚偣锛氫负浠€涔圓OI璁惧鎬绘妸鐒婄偣褰撳垝鐥曪紵

鑰佸紶鏈€杩戝揩琚€佹澘楠傚偦浜嗏€斺€斾粬浠敤YOLOv5鍋氱殑鐢佃矾鏉胯川妫€绯荤粺锛屾妸0.5mm鐨勭數瀹圭己浠舵娴嬫垚姝e父锛屽嵈鎬绘妸鍙嶅厜鐨勭剨鐐瑰垽涓哄垝鐥曘€傜湅鐫€浜х嚎姣忓ぉ2000鍧楃數璺澘闇€瑕佷汉宸ュ妫€锛岃繖浜嬪繀椤诲緱瑙e喅锛?/p>


馃暤锔忊檪锔忕牬妗堟€濊矾锛氫笂閲囨牱瀵艰嚧鐨勭壒寰佹贩娣嗭紙H3锛?/h3>

鎷嗗紑妯″瀷杈撳嚭鐗瑰緛鍥句竴鐪嬶紝濂藉浼欙紒3x3鐨勭己浠剁洰鏍囧湪32鍊嶄笅閲囨牱鍚庡彧鍓╀釜鍍忕礌鐐广€傝繖鏃跺€欑敤榛樿鐨勨€?strong>鈥嬫渶杩戦偦涓婇噰鏍封€?/strong>鈥嬶紝灏卞儚鐢ㄩ┈璧涘厠鐢荤數璺浘鈥斺€旂粏鑺傚叏绯婁簡銆?/p>

馃挕鍏抽敭鍙戠幇锛?/p>

  1. 宸ヤ笟寰皬鐩爣闇€瑕佲€?strong>鈥嬩簹鍍忕礌绾ц繕鍘熲€?/strong>鈥?/li>
  2. 鍙嶅厜骞叉壈瑕佹眰鈥?strong>鈥嬭竟缂橀噸寤烘洿绮惧噯鈥?/strong>鈥?/li>
  3. 浜х嚎瀹炴椂鎬ч渶瑕佲€?strong>鈥嬫帶鍒惰绠楄€楁椂鈥?/strong>鈥?/li>

馃洜锔忓洓姝ヨ皟浼樻柟妗堬紙甯﹀弬鏁版ā鏉匡級

鈥?strong>鈥嬬鈶犳锛氭浛鎹笂閲囨牱鏂规硶鈥?/strong>鈥?br/> 鎶妝olov5s.yaml閲岀殑锛?/p>

python澶嶅埗
[-1, 1, nn.Upsample, [None, 2, 'nearest']]

鏀规垚锛?/p>

python澶嶅埗
[-1, 1, nn.ConvTranspose2d, [128, 4, 2, 1, 0, bias=False]]

鈥?strong>鈥嬬鈶℃锛氱壒寰佽瀺鍚堝寮衡€?/strong>鈥?br/> 鍦℉ead灞傚墠鎻掑叆锛?/p>

python澶嶅埗
[from_layer1, from_layer2, 1, CBAM, []]  # 绌洪棿娉ㄦ剰鍔涙ā鍧?/span>

鈥?strong>鈥嬬鈶㈡锛氬姩鎬佺缉鏀剧瓥鐣モ€?/strong>鈥?br/> 鏍规嵁杈撳叆灏哄鑷姩璋冩暣涓婇噰鏍峰€嶆暟锛?/p>

python澶嶅埗
scale_factor = max(1, img_size//640)  # 4K鍥惧儚鑷姩鍗囨。

鈥?strong>鈥嬬鈶f锛氭姉閿娇鍚庡鐞嗏€?/strong>鈥?br/> 杈撳嚭灞傚墠澧炲姞锛?/p>

python澶嶅埗
[1, 1, nn.Conv2d, [256, 3, 1, 1], {'padding':1}]

馃彮钀藉湴鏁堟灉瀵规瘮锛堝惈鐢熶骇鏁版嵁锛?/h3>
鎸囨爣浼樺寲鍓?/th>浼樺寲鍚?/th>鎻愬崌骞呭害
缂轰欢妫€鍑虹巼68.2%93.7%+25.5%
璇姤鐜?/td>22.1%5.3%-76%
鎺ㄧ悊閫熷害38ms45ms+18%
GPU鍗犵敤2.8GB3.1GB+10.7%

馃搶浜х嚎鐪熷疄鏁版嵁锛氭繁鍦虫煇鐢靛瓙鍘傚簲鐢ㄥ悗锛屾瘡鏈堝噺灏戜汉宸ュ妫€鎴愭湰12涓囧厓锛岃鍒ゅ鑷寸殑鐗╂枡鎹熻€椾笅闄?4%


馃毇閬垮潙鎸囧崡锛氳繖浜涙搷浣滃崈涓囧埆璇曪紒

  1. 鐩茬洰浣跨敤PixelShuffle锛氭煇瀹㈡埛寮鸿涓奝S瀵艰嚧3090鏄惧崱鐖嗘樉瀛?/li>
  2. 鍫嗗彔澶氬眰涓婇噰鏍凤細鏌愭柟妗堝晢鎼?灞備笂閲囨牱锛屾帹鐞嗗欢杩熼鍒?00ms+
  3. 蹇界暐杈撳叆鍒嗚鲸鐜囷細4K鍥惧儚鐢?40璁粌鐩存帴涓婇噰鏍凤紝鏁堟灉鍫瘮鐏甸瓊鐢绘墜

馃挕鐙鍔ㄦ€佷笂閲囨牱绛栫暐

鎴戜滑鑷爺鐨勨€?strong>鈥嬪脊鎬х缉鏀炬ā鍧椻€?/strong>鈥嬶紙浠g爜鏆備笉寮€婧愶級鑳芥牴鎹洰鏍囧瘑搴﹁嚜鍔ㄨ皟鑺傦細

  • 褰撴娴嬪埌>10涓?cm虏鐨勫瘑闆嗙洰鏍囨椂锛屽惎鐢ㄥ弻璺笂閲囨牱骞惰仈
  • 鐩爣绋€鐤忔椂鑷姩鍒囨崲杞婚噺妯″紡
    瀹炴祴鍦ㄨ姱鐗囧紩鑴氭娴嬩腑锛宮AP鎻愬崌9.8%鐨勫悓鏃讹紝鎺ㄧ悊閫熷害浠呴檷浣?ms

馃鍐欑粰宸ョ▼甯堢殑鏆磋

涓婇噰鏍蜂紭鍖栧氨鍍忕粰妯″瀷閰嶇溂闀溾€斺€?/p>

  • 鍙岀嚎鎬ф彃鍊兼槸50鍧楄€佽姳闀?/li>
  • 杞疆鍗风Н鏄畾鍒舵笎杩涘鐒﹂暅鐗?/li>
  • 鍔ㄦ€佺瓥鐣ュ氨鏄櫤鑳藉彉鐒﹀ⅷ闀?/li>

涓嬫閬囧埌妫€娴嬮毦棰橈紝鍒€ョ潃鎹㈡ā鍨嬶紝鍏堢粰妯″瀷閰嶅壇濂?鐪奸暅"鍚э紒

本文由嘻道妙招独家原创,未经允许,严禁转载