資源簡介
該文檔內(nèi)包含有DenseNet 實(shí)現(xiàn)以及Attention Unet網(wǎng)絡(luò)結(jié)構(gòu)的Pytorch實(shí)現(xiàn)，其中使用到dice loss，交叉熵loss以及基于focal loss思想改造后的適用于pixel級(jí)別分類的pixel focal loss（在test loss里面），這是項(xiàng)目的完整文件，包含有整個(gè)完整的參數(shù)設(shè)置、訓(xùn)練、測(cè)試流程以及相應(yīng)的可視化過程
代碼片段和文件信息
#?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（） ?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????文件?????????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個(gè)文件信息
評(píng)論
共有 條評(píng)論