资源简介
该文档内包含有DenseNet 实现以及Attention Unet网络结构的Pytorch实现，其中使用到dice loss，交叉熵loss以及基于focal loss思想改造后的适用于pixel级别分类的pixel focal loss（在test loss里面），这是项目的完整文件，包含有整个完整的参数设置、训练、测试流程以及相应的可视化过程
代码片段和文件信息
# coding:utf-8
class Config_test（object）:
def __init__（self）:
self.test_root_path = r‘G:\Data\BreastData\BreastMass‘
self.test_src_tag = ‘test\src‘
self.test_filelist = r“G:\Data\BreastData\BreastMass\test\test.txt“
self.savepath = r‘G:\Data\BreastData\BreastMass\test\mask_result0422‘
self.test_shape = （1024 1024） # should be the same as train_shape
self.test_checkpoint = r“G:\Code\DenseNet_Segmentation\checkpoint\AttU_Net_20190422_075756.pt“
self.device_ids = ‘0‘
self.batch_size = 1
self.workers = 1
self.mask_th = 0.5 # probility threshold
test_cf = Config_test（） 属性 大小 日期 时间 名称
----------- --------- ---------- ----- ----
文件 688 2019-04-22 10:05 config_test.py
文件 1262 2019-04-21 15:05 config_train.py
目录 0 2019-04-17 11:38 datasets\
文件 1964 2019-04-17 10:18 datasets\DataSets.py
文件 14419 2019-04-17 11:38 datasets\utils.py
文件 40 2019-03-27 10:30 datasets\__init__.py
目录 0 2019-04-19 10:36 Loss\
文件 3911 2019-04-16 23:27 Loss\diceloss.py
文件 6124 2019-04-16 09:18 Loss\discriminative.py
文件 3545 2019-04-17 16:14 Loss\focalloss.py
文件 7811 2019-04-16 18:04 Loss\lovaloss.py
文件 2138 2019-04-18 13:12 Loss\testloss.py
文件 1285 2019-04-19 10:36 Loss\totalloss.py
文件 0 2019-04-15 15:51 Loss\__init__.py
目录 0 2019-04-19 11:25 models\
文件 6127 2019-04-17 11:20 models\g_cam.py
文件 3137 2019-03-25 09:58 models\la
文件 18368 2019-04-19 11:25 models\model.py
目录 0 2019-04-19 11:34 models\runs\
目录 0 2019-04-19 11:27 models\runs\Apr19_11-27-52_DESKTOP-F5GFDSAAttU_Net\
文件 25 2019-04-19 11:27 models\runs\Apr19_11-27-52_DESKTOP-F5GFDSAAttU_Net\events.out.tfevents.1555644472.DESKTOP-F5GFDSA
目录 0 2019-04-19 11:31 models\runs\Apr19_11-31-06_DESKTOP-F5GFDSAAttU_Net\
文件 25 2019-04-19 11:31 models\runs\Apr19_11-31-06_DESKTOP-F5GFDSAAttU_Net\events.out.tfevents.1555644666.DESKTOP-F5GFDSA
目录 0 2019-04-19 11:32 models\runs\Apr19_11-32-30_DESKTOP-F5GFDSAAttU_Net\
文件 25 2019-04-19 11:32 models\runs\Apr19_11-32-30_DESKTOP-F5GFDSAAttU_Net\events.out.tfevents.1555644750.DESKTOP-F5GFDSA
目录 0 2019-04-19 11:34 models\runs\Apr19_11-34-10_DESKTOP-F5GFDSAAttU_Net\
文件 25 2019-04-19 11:34 models\runs\Apr19_11-34-10_DESKTOP-F5GFDSAAttU_Net\events.out.tfevents.1555644850.DESKTOP-F5GFDSA
文件 2246 2019-04-16 11:54 models\tools.py
文件 122 2019-03-25 10:10 models\__init__.py
文件 2878 2019-04-17 11:42 test.py
文件 7216 2019-04-17 12:04 train.py
............此处省略3个文件信息
评论
共有 条评论