原创 mxnet 解析模型的參數

這段時間的工作一直在圍繞移動端展開。項目需要在手機上跑一個深度學習模型,所以直接上resnet或者是其他的比較好的重量級的模型是不現實的,甚至mobilenet都是不太理想的,我用的一個千元機,跑mobilenet幾乎需要1秒的時間。所以

原创 Mxnet 實現自己的dataiter

實現深度學習模型的時候,有時候dataiter會不能滿足自己的需求,所以需要繼承下來,自己寫一下。mxnet寫customed dataiter代碼也比較簡單,直接上代碼了import mxnet as mx class custom_

原创 mxnet實現自己的圖像數據集增強方法

        深度學習做圖像相關的內容時候,數據集增強是常用並且十分有效的手段,可以有效的對口過擬合以及提高模型的準確率,針對不同的問題有時候需要特定的方式對數據進行變換。Mxnet已經內置了一些常用的增強手段,例如randomcrop