-
大小: 5KB文件類(lèi)型: .zip金幣: 1下載: 0 次發(fā)布日期: 2021-01-06
- 語(yǔ)言: 其他
- 標(biāo)簽: pytorch??分類(lèi)網(wǎng)絡(luò)??
資源簡(jiǎn)介
使用pytorch搭建分類(lèi)網(wǎng)絡(luò)，針對(duì)Luna16數(shù)據(jù)集生成的疑似肺結(jié)節(jié)進(jìn)行分類(lèi)，實(shí)現(xiàn)假陽(yáng)性剔除。其樣本集保存為.Mat的形式（數(shù)據(jù)+類(lèi)別），固定大小為24*40*40。
注意!!!!!原始getMat.py、traindataset.py有誤（有bug），因?yàn)閏sdn無(wú)法修改資源，詳情見(jiàn)https://blog.csdn.net/qq_24739717/article/details/101034728
注意!!!!!原始getMat.py、traindataset.py有誤（有bug），因?yàn)閏sdn無(wú)法修改資源，詳情見(jiàn)https://blog.csdn.net/qq_24739717/article/details/101034728
代碼片段和文件信息
import?scipy.io?as?sio
import?numpy?as?np
import?os
import?random?
import?torch
from?torch.utils.data?import?DataLoader
from?torchvision?import?datasets
from?torchvision?import?transforms
from?torch.autograd?import?Variable
import?torch.optim?as?optim
import?torch.nn.functional?as?F
import?argparse
#模型
from?model?import?*
#數(shù)據(jù)
from?traindataset?import?*
if?__name__?==?“__main__“:
????#設(shè)置參數(shù)
????parser?=?argparse.ArgumentParser（）
????parser.add_argument（“--epochs“?type=int?default=80?help=“number?of?epochs“）
????parser.add_argument（“--batch_size“?type=int?default=64?help=“size?of?each?image?batch“）
????parser.add_argument（“--gradient_accumulations“?type=int?default=2?help=“number?of?gradient?accums?before?step“）
????parser.add_argument（“--n_cpu“?type=int?d?屬性????????????大小?????日期????時(shí)間???名稱(chēng)
-----------?---------??----------?-----??----
?????文件????????6448??2019-09-19?21:04??Pytorch分類(lèi)網(wǎng)絡(luò)（肺結(jié)節(jié)假陽(yáng)性剔除）\classfierMat.py
?????文件????????4811??2019-09-17?14:23??Pytorch分類(lèi)網(wǎng)絡(luò)（肺結(jié)節(jié)假陽(yáng)性剔除）\model.py
?????文件????????2584??2019-09-19?21:03??Pytorch分類(lèi)網(wǎng)絡(luò)（肺結(jié)節(jié)假陽(yáng)性剔除）\traindataset.py
?????目錄???????????0??2019-09-19?20:33??Pytorch分類(lèi)網(wǎng)絡(luò)（肺結(jié)節(jié)假陽(yáng)性剔除）\
評(píng)論
共有 條評(píng)論