国产探花免费观看_亚洲丰满少妇自慰呻吟_97日韩有码在线_资源在线日韩欧美_一区二区精品毛片,辰东完美世界有声小说,欢乐颂第一季,yy玄幻小说排行榜完本

首頁 > 編程 > Python > 正文

caffe的python接口學習(5):生成deploy文件

2019-11-06 08:25:05
字體:
來源:轉載
供稿:網友

如果要把訓練好的模型拿來測試新的圖片,那必須得要一個deploy.PRototxt文件,這個文件實際上和test.prototxt文件差不多,只是頭尾不相同而也。deploy文件沒有第一層數據輸入層,也沒有最后的Accuracy層,但最后多了一個Softmax概率層。

這里我們采用代碼的方式來自動生成該文件,以mnist為例。

deploy.py

復制代碼
# -*- coding: utf-8 -*-from caffe import layers as L,params as P,to_protoroot='/home/xxx/'deploy=root+'mnist/deploy.prototxt'    #文件保存路徑def create_deploy():    #少了第一層,data層    conv1=L.Convolution(bottom='data', kernel_size=5, stride=1,num_output=20, pad=0,weight_filler=dict(type='xavier'))    pool1=L.Pooling(conv1, pool=P.Pooling.MAX, kernel_size=2, stride=2)    conv2=L.Convolution(pool1, kernel_size=5, stride=1,num_output=50, pad=0,weight_filler=dict(type='xavier'))    pool2=L.Pooling(conv2, pool=P.Pooling.MAX, kernel_size=2, stride=2)    fc3=L.InnerProduct(pool2, num_output=500,weight_filler=dict(type='xavier'))    relu3=L.ReLU(fc3, in_place=True)    fc4 = L.InnerProduct(relu3, num_output=10,weight_filler=dict(type='xavier'))    #最后沒有accuracy層,但有一個Softmax層    prob=L.Softmax(fc4)    return to_proto(prob)def write_deploy():     with open(deploy, 'w') as f:        f.write('name:"Lenet"/n')        f.write('input:"data"/n')        f.write('input_dim:1/n')        f.write('input_dim:3/n')        f.write('input_dim:28/n')        f.write('input_dim:28/n')        f.write(str(create_deploy()))if __name__ == '__main__':    write_deploy()復制代碼

運行該文件后,會在mnist目錄下,生成一個deploy.prototxt文件。

這個文件不推薦用代碼來生成,反而麻煩。大家熟悉以后可以將test.prototxt復制一份,修改相應的地方就可以了,更加方便。


發表評論 共有條評論
用戶名: 密碼:
驗證碼: 匿名發表
主站蜘蛛池模板: 藁城市| 东乡族自治县| 宁波市| 钟山县| 砚山县| 陈巴尔虎旗| 诸城市| 清流县| 五河县| 清远市| 探索| 措勤县| 广平县| 于都县| 陈巴尔虎旗| 保山市| 衡阳县| 鹿邑县| 左贡县| 万安县| 禹州市| 灵山县| 确山县| 兰州市| 宁武县| 贵定县| 高州市| 宾阳县| 杨浦区| 肇州县| 吉木萨尔县| 景德镇市| 南陵县| 宝鸡市| 房产| 余干县| 常德市| 蓬溪县| 鄯善县| 武川县| 东安县|