Caffe 實現多標籤分類 支持Multi-Label的LMDB數據格式輸入

Caffe自帶的圖像轉LMDB接口只支持單label,對於多label的任務,可以使用HDF5的格式,也可以通過修改caffe代碼來實現, 我的文章Caffe 實現多標籤分類 裏介紹了怎麼通過修改ImageDataLayer來實現Multilabel的任務, 本篇文章介紹怎麼通過修改DataLayer來實現帶Multilabel的LMDB格式數據輸入的分類任務 1. 首先修改代碼        
相關文章
相關標籤/搜索