資源簡介
使用pytorch搭建分類網絡，針對Luna16數據集生成的疑似肺結節進行分類，實現假陽性剔除。其樣本集保存為.Mat的形式（數據+類別），固定大小為24*40*40。
注意!!!!!原始getMat.py、traindataset.py有誤（有bug），因為csdn無法修改資源，詳情見https://blog.csdn.net/qq_24739717/article/details/101034728
注意!!!!!原始getMat.py、traindataset.py有誤（有bug），因為csdn無法修改資源，詳情見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?*
#數據
from?traindataset?import?*
if?__name__?==?“__main__“:
????#設置參數
????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?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????文件????????6448??2019-09-19?21:04??Pytorch分類網絡（肺結節假陽性剔除）\classfierMat.py
?????文件????????4811??2019-09-17?14:23??Pytorch分類網絡（肺結節假陽性剔除）\model.py
?????文件????????2584??2019-09-19?21:03??Pytorch分類網絡（肺結節假陽性剔除）\traindataset.py
?????目錄???????????0??2019-09-19?20:33??Pytorch分類網絡（肺結節假陽性剔除）\
評論
共有 條評論