Pytorch分类网络肺结节假阳性剔除.zip

使用pytorch搭建分类网络，针对Luna16数据集生成的疑似肺结节进行分类，实现假阳性剔除。其样本集保存为.Mat的形式（数据+类别），固定大小为24*40*40。
注意!!!!!原始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分类网络（肺结节假阳性剔除）\