Python影像辨識筆記(二十七):如何針對YOLOv3/v4進行剪枝(Pruning)?

卷積網路剪枝論文

其他有關 Pruning的介紹

YOLOv3/v4實作教學

建議透過以下兩篇文章理解剪枝程式的撰寫方式,但不採用這兩篇內文所提供的程式碼。原因在於作者所提供的程式碼似乎只適用於YOLOv3,我自己透過YOLOv4實測發現似乎會有一些Bug存在。

建議使用的程式碼

Machine Learning / Deep Learning / Python / Flutter cakeresume.com/yanwei-liu

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store