-
大小: 9KB文件類型: .zip金幣: 1下載: 0 次發布日期: 2021-01-08
- 語言: 其他
- 標簽: cnn??圖像分類??tensorflow??
資源簡介
這是利用卷積神經網絡進行貓和狗的圖像分類的相關代碼,由于訓練和測試的數據集較大,所以沒有放上來,訓練數據集需要自行下載
代碼片段和文件信息
import?tensorflow?as?tf
import?numpy?as?np
import?os
‘‘‘生成圖片路徑和標簽的List‘‘‘
def?get_files(file_dir):
????“““
???????輸入:?存放訓練照片的文件地址
???????返回:??圖像列表,?標簽列表
???????“““
????cats?=?[]
????label_cats?=?[]
????dogs?=?[]
????label_dogs?=?[]
????for?file?in?os.listdir(file_dir):??#?用os.listdir函數來返回file_dir路徑下所有的圖片,file就是要讀取的照片
????????name?=?file.split(sep=‘.‘)??#?用file.split?將文件的名字分隔開,分隔的符號是‘.’
????????if?name[0]?==?‘cat‘:??#?所以只用讀取?.?前面這個字符串
????????????cats.append(file_dir?+?file)
????????????label_cats.append(0)??#?把圖像和標簽加入列表
????????else:
????????????dogs.append(file_dir?+?file)
????????????label_dogs.append(1)??#?名字是cat就賦值0,名字是dog就賦值1
????print(‘There?are?%d?cats\nThere?are?%d?dogs‘?%?(len(cats)?len(dogs)))#打印有多少貓,多少狗
????‘‘‘對生成的圖片路徑和標簽List做打亂處理‘‘‘
????image_list?=?np.h
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????文件????????5327??2019-10-21?16:55??代碼\input_data.py
?????文件???????10635??2019-10-21?09:34??代碼\model.py
?????文件????????2679??2019-10-21?17:06??代碼\test.py
?????文件????????3677??2019-10-21?16:41??代碼\training.py
?????目錄???????????0??2020-03-19?10:17??代碼\
- 上一篇:網絡調試助手最新版本4.13版本
- 下一篇:計算機圖形學 立方體動態畫家隱面算法
評論
共有 條評論