CANet復(fù)現(xiàn)
環(huán)境配置
首先在Anaconda中創(chuàng)建虛擬中創(chuàng)建新的虛擬環(huán)境
conda create -n env_name python=X.X
conda info -e
env_name 是該虛擬環(huán)境的名字，可以命名為CANet X.X選擇python版本，這里原論文作者使用的是python3.7，所以在選擇版本時要選擇至少3.7以上的版本
創(chuàng)建好虛擬環(huán)境后我們可以用下列命令查看我們所有環(huán)境的情況
查看環(huán)境
用下列命令激活虛擬環(huán)境
conda activate env_name
如圖所示，激活虛擬環(huán)境CANet
激活虛擬環(huán)境
對虛擬環(huán)境進(jìn)行配置，通過pip命令來下載所需要的包
pip freeze > requirements.txt
這個時候在上面的路經(jīng)下就會生成一個名為requirements的txt文件
生成文件
找到文件
找到這個文件，進(jìn)行如下修改
astroid==2.4.2
certifi==2021.10.8
colorama==0.4.4
isort==5.6.4
lazy-object-proxy==1.4.3
mccabe==0.6.1
pylint==2.6.0
six==1.15.0
toml==0.10.2
typed-ast==1.4.1
wincertstore==0.2
wrapt==1.12.1
certifi==2021.10.8
charset-normalizer==2.0.12
colorama==0.4.4
cycler==0.11.0
fonttools==4.32.0
idna==3.3
imageio==2.17.0
jsonpatch==1.32
jsonpointer==2.3
kiwisolver==1.4.2
matplotlib==3.5.1
networkx==2.6
nibabel==3.2.2
numpy==1.21.6
opencv-python==4.5.5.64
packaging==21.3
pandas==1.3.2
Pillow==9.1.0
pyparsing==3.0.8
python-dateutil==2.8.2
pytz==2022.1
PyWavelets==1.3.0
pyzmq==22.3.0
requests==2.27.1
scikit-image==0.19.2
scipy==1.7.3
tifffile==2021.11.2
torch==1.11.0
torchaudio==0.11.0
torchfile==0.1.0
torchvision==0.12.0
tornado==6.1
tqdm==4.64.0
typing_extensions==4.2.0
urllib3==1.26.9
visdom==0.1.8.9
websocket-client==1.3.2
wincertstore==0.2
如有報錯調(diào)整版本號即可
下載好所需要的包后配置完成
數(shù)據(jù)集處理
在github上下載的zip包里面解壓后，有一個readme.md，打開后查看Requirementss，Usages兩個部分，找到數(shù)據(jù)下載的鏈接
下載數(shù)據(jù)集
按照Usages里面的步驟進(jìn)行數(shù)據(jù)預(yù)處理
處理數(shù)據(jù)集
本地數(shù)據(jù)預(yù)處理
根據(jù)自己的路經(jīng)修改root_dir
執(zhí)行五倍的交叉驗證
屏幕截圖 2022-04-25 173739.png
在終端里訓(xùn)練模型
python main.py --data ISIC2018 --val_folder folder1 --id Comp_Atten_Unet --batch_size 2 --epochs 5
屏幕截圖 2022-04-25 201923.png
評估模型
python validation.py --data ISIC2018 --val_folder folder0 --id Comp_Atten_Unet
訓(xùn)練完畢
評估模型
輸出heat map
heat map
