深度殘差收縮網絡之五:代碼實現

深度殘差收縮網絡其實是一種通用的特徵學習方法,是深度殘差網絡ResNet、注意力機制和軟閾值化的集成,可以用於圖像分類。本文采用TensorFlow 1.0和TFLearn 0.3.2,編寫了圖像分類的程序,採用的圖像數據爲CIFAR-10。CIFAR-10是一個非常常用的圖像數據集,包含10個類別的圖像。可以在這個網址找到具體介紹:https://www.cs.toronto.edu/~kriz
相關文章
相關標籤/搜索