測(cè)試項(xiàng)目為CycleGAN
項(xiàng)目源地址:https://github.com/xhujoy/CycleGAN-tensorflow
百度云地址:鏈接:https://pan.baidu.com/s/1tT1mUBqpFiN0kbEVvXlJjg 提取碼:6n3r 里面有部分?jǐn)?shù)據(jù)集
一:安裝Anaconda
步驟:
1:從官方網(wǎng)站下載Anaconda
https://www.anaconda.com/download/
2:進(jìn)行軟件安裝(這個(gè)和普通的沒什么特別區(qū)別)
注意一點(diǎn):
3:安裝完成Anaconda之后進(jìn)行環(huán)境變量的測(cè)試
進(jìn)入到windows中的CMD命令模式:
(1)檢測(cè)anaconda環(huán)境是否安裝成功:conda --version
version的版本號(hào)根據(jù)自己下載的變化月而,并不影響
(2)檢測(cè)目前安裝了哪些環(huán)境變量:conda info --envs
4:更改源
更改方法一:cmd后依次輸入下面命令
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
二、創(chuàng)建anaconda環(huán)境
1:創(chuàng)建環(huán)境讨盒, 安裝python
python版本(我這里是安裝的3.6的版本是嗜,這個(gè)根據(jù)需求來吧):
conda create --name tensorflow-gpu python=3.6 #tensorflow-gpu為環(huán)境名字愈案,
2:激活tensorflow 環(huán)境
conda activate tensorflow-gpu #tensorflow-gpu為自己剛才創(chuàng)建的名字
這里括號(hào)中表示激活的環(huán)境
3:anaconda環(huán)境的其他操作(刪除,重命名鹅搪,拷貝)可自行百度
三站绪、安裝pycharm
Pycharm是程序的編輯器
下載地址:https://www.jetbrains.com/pycharm/download/#section=windows
社區(qū)版不需要激活涩嚣,專業(yè)版需要激活(可以使用自己的湖大郵箱注冊(cè)賬號(hào)航厚,進(jìn)行學(xué)生認(rèn)證幔睬,對(duì)學(xué)生免費(fèi)麻顶,當(dāng)然如果懶得搞,就下載社區(qū)版)
一路路燈安裝就行
四矫钓、安裝tensorflow
1:CPU版本的安裝
可以參考價(jià)官網(wǎng)安裝
https://www.tensorflow.org/install/pip?hl=zh-cn
在激活了虛擬環(huán)境之后
在CMD中輸入
pip install --upgrade tensorflow==1.15
本次代碼建議使用1.15版本,不要問為什么,就因?yàn)槲易约菏沁@個(gè)版本饲握,任性救欧。不要裝2.0版本之后的,因?yàn)?.0版本跟1.X版本相差很大寄猩,很多庫不一樣田篇。
2:GPU版本安裝
1: 必須在系統(tǒng)中安裝以下 NVIDIA? 軟件:
- NVIDIA? GPU 驅(qū)動(dòng)程序
- CUDA? 工具包
- CUDA? 工具包附帶的 CUPTI。
- cuDNN SDK 8.1.0 cuDNN 版本。
cuda CUDNN tensorflow版本一定要對(duì)應(yīng)W创稹!馆截!
cuda CUDNN tensorflow版本一定要對(duì)應(yīng)!=颜拧宿接!
cuda CUDNN tensorflow版本一定要對(duì)應(yīng)!J嗯狻!
對(duì)應(yīng)關(guān)系可以參考:
2:安裝驅(qū)動(dòng)
在英偉達(dá)官網(wǎng)安裝或者英偉達(dá)控制面板中更新到最新
3:安裝cuda和CUDNN
1)下載cuda
CUDA下載地址:
https://developer.nvidia.com/cuda-toolkit-archive
2)下載cudnn
CUDNN下載地址:
https://developer.nvidia.com/zh-cn/cudnn
下載cudnn需要注冊(cè)登陸踏拜。一定要一定要一定要匹配版本肮塞。
3)安裝cuda
一路綠燈姻锁,一路默認(rèn)。
4)安裝cudnn
將cudnn的文件解壓縮會(huì)得到這三個(gè)東西
然后講這三個(gè)文件夾中的東西拷貝到cuda安裝路徑中相對(duì)應(yīng)的位置如果是默認(rèn)安裝就會(huì)裝到這里。
測(cè)試tensorflow
在CMD中輸入這些,如果沒報(bào)錯(cuò)就恭喜了@好痢8尘!!
4.TensorFlow測(cè)試
(1).代碼測(cè)試:
import tensorflow as tf
如果安裝過程中出現(xiàn)問題懊昨,直接復(fù)制問題到百度窄潭,基本都有解釋
參考:https://blog.csdn.net/sb19931201/article/details/53648615
五、打開項(xiàng)目
1:解壓縮文件酵颁。
2:打開pycharm嫉你,打開一個(gè)項(xiàng)目,路徑就是剛才解壓的路徑
3:設(shè)置環(huán)境躏惋,設(shè)置為剛才創(chuàng)建和安裝好的的tensorflow環(huán)境
六幽污、安裝依賴
每個(gè)程序在開始都有導(dǎo)入的庫,如果庫沒有安裝在運(yùn)行程序會(huì)報(bào)錯(cuò)提示沒有這個(gè)東西其掂,
安裝庫一般可以在pycharm 下方terminal選型卡進(jìn)行pip安裝油挥,以安裝argparse庫為例
使用pip install argparse(要安裝的包)
這里軟件所需要的的包已經(jīng)在requirement.txt文件中記錄了。我們只需要使用
pip install -r requirements.txt
即可安裝款熬,如果遇到個(gè)別的包沒安裝正確,可以使用下面命令來安裝攘乒。這個(gè)例子中scipy1.2.1要和PILLOW==7.0.0要對(duì)應(yīng)贤牛,如果有問題很大可能是這個(gè)問題
pip install scipy==1.2.1 #指定版本號(hào)需要使用兩個(gè)等號(hào),
七则酝、代碼解釋
1各文件的作用
1:checkpoint 這個(gè)文件夾放置訓(xùn)練好的模型
2:datasets 文件夾放置數(shù)據(jù)集殉簸,數(shù)據(jù)路徑在圖中有顯示
3:img 文件夾用來放置readme.md中的圖片,
4:sample 文件夾用來放置訓(xùn)練時(shí)產(chǎn)生的樣例
5:test 文件夾用來放置模型訓(xùn)練好之后的測(cè)試結(jié)果
6:download——dataset.sh 數(shù)據(jù)集下載腳本沽讹。在Ubuntu系統(tǒng)可以直接下載般卑,本文例子直接給了數(shù)據(jù)集。
7:main.py 主程序爽雄,程序從這里開始運(yùn)行
8:model.py 定義cyclegan 模型
9:module.py 用來定義cyclegan的基本結(jié)構(gòu)蝠检,如生成器鑒別器
10:ops.py:定義生成器和鑒別器中的卷積層
11:utils.py:定義cyclegan中的工具。
2模型定義代碼
1)訓(xùn)練參數(shù)
2)組件定義
3)cyclegan模型定義
定義cycle結(jié)構(gòu)挚瘟,和鑒別器的判別結(jié)果
4)損失定義
5)優(yōu)化方式定義