update weights

FBK-NILab · Apr 22, 2022 · 9922e20 · 9922e20
1 parent 87b5cc7
commit 9922e20
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/UnetBrainSeg.py b/UnetBrainSeg.py
@@ -52,7 +52,7 @@ def init_unet(checkpoint_dir=None,checkpoint_basename=None,ckpt_step=None,checkp
        checkpoint_basename = ext_name_train
 
    if ckpt_step is None:
-       ckpt_step="24221"
+       ckpt_step="24274"#"24111"#"25021"#"24467"#"24202"#"24221"
    checkpoint_file1=checkpoint_dir+ "/"+checkpoint_basename+"-"+ckpt_step+".data-00000-of-00001"
    checkpoint_file2=checkpoint_dir+ "/"+checkpoint_basename+"-"+ckpt_step+".index"
    checkpoint_file3=checkpoint_dir+ "/"+checkpoint_basename+"-"+ckpt_step+".meta"
@@ -62,11 +62,13 @@ def init_unet(checkpoint_dir=None,checkpoint_basename=None,ckpt_step=None,checkp
 
 	trymakedir(checkpoint_dir)
 
-	gdd.download_file_from_google_drive(file_id='10E1PhOazB8RDx9pPDFCXKtl5mgylqZm-',
+
+
+	gdd.download_file_from_google_drive(file_id='1hcm_LIDsmtnopkK7380E9xr8RgFkjkMa-',
                                     dest_path=checkpoint_file1)
-	gdd.download_file_from_google_drive(file_id='1KBk5kWoyhNvE0iMeM25cqrzUSoSwQHYH',
+	gdd.download_file_from_google_drive(file_id='10AfOmscVoUiiu01s_v1VpC05gXnR1k3w',
                                     dest_path=checkpoint_file2)
-	gdd.download_file_from_google_drive(file_id='1dwbg4uZpXWk7WXzxn7zWpdtVxxgKQSbR',
+	gdd.download_file_from_google_drive(file_id='1DU3e1SKHQhpDTzPAwVXzrpiolhsiiMa9',
                                     dest_path=checkpoint_file3)	
    checkpoint_file=checkpoint_dir+ "/checkpoint"
 

diff --git a/...256x256x256_filter12_dp0.15_sparse_softmax_cross_entropy/DisSeg-24274.data-00000-of-00001 b/...256x256x256_filter12_dp0.15_sparse_softmax_cross_entropy/DisSeg-24274.data-00000-of-00001
diff --git a/...points/DisSeg_256x256x256_filter12_dp0.15_sparse_softmax_cross_entropy/DisSeg-24274.index b/...points/DisSeg_256x256x256_filter12_dp0.15_sparse_softmax_cross_entropy/DisSeg-24274.index
diff --git a/...kpoints/DisSeg_256x256x256_filter12_dp0.15_sparse_softmax_cross_entropy/DisSeg-24274.meta b/...kpoints/DisSeg_256x256x256_filter12_dp0.15_sparse_softmax_cross_entropy/DisSeg-24274.meta