View GitHub Repository DeepReg

Schedule Inspection
Subscribe
Passed
Pull Request — main (#675)
by Yunguan
03:11
created

classical_ct_headneck_affine.demo_data   A

↳ Parent: Project

Complexity

Total Complexity 0

Size/Duplication

Total Lines 25
Duplicated Lines 0 %

Importance

Changes 0
Metric Value
wmc 0
eloc 16
dl 0
loc 25
rs 10
c 0
b 0
f 0
1 
"""
2 
Download the demo data
3 
"""
4 
import os
5 
import shutil
6 

            


            
                                                        
            
                                    
            
            

                
                    7
                

                                    
                                                    
                
                
                
from tensorflow.keras.utils import get_file

            


            
                                                        
            
                                    
            
            

                
                    8
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    9
                

                                    
                                                    
                
                
                
MAIN_PATH = os.getcwd()

            


            
                                                        
            
                                    
            
            

                
                    10
                

                                    
                                                    
                
                
                
PROJECT_DIR = "demos/classical_ct_headneck_affine"

            


            
                                                        
            
                                    
            
            

                
                    11
                

                                    
                                                    
                
                
                
os.chdir(PROJECT_DIR)

            


            
                                                        
            
                                    
            
            

                
                    12
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    13
                

                                    
                                                    
                
                
                
DATA_PATH = "dataset"

            


            
                                                        
            
                                    
            
            

                
                    14
                

                                    
                                                    
                
                
                
FILE_PATH = os.path.abspath(os.path.join(DATA_PATH, "demo.h5"))

            


            
                                                        
            
                                    
            
            

                
                    15
                

                                    
                                                    
                
                
                
ORIGIN = "https://github.com/YipengHu/example-data/raw/master/hnct/demo.h5"

            


            
                                                        
            
                                    
            
            

                
                    16
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    17
                

                                    
                                                    
                
                
                
if os.path.exists(DATA_PATH):

            


            
                                                        
            
                                    
            
            

                
                    18
                

                                    
                                                    
                
                
                
    shutil.rmtree(DATA_PATH)

            


            
                                                        
            
                                    
            
            

                
                    19
                

                                    
                                                    
                
                
                
os.mkdir(DATA_PATH)

            


            
                                                        
            
                                    
            
            

                
                    20
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    21
                

                                    
                                                    
                
                
                
get_file(FILE_PATH, ORIGIN)

            


            
                                                        
            
                                    
            
            

                
                    22
                

                                    
                                                    
                
                
                
print("CT head-and-neck data downloaded: %s." % FILE_PATH)

            


            
                                                        
            
                                    
            
            

                
                    23
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    24
                

                                    
                                                    
                
                
                
os.chdir(MAIN_PATH)

            


            
                                                        
            
                                    
            
            

                
                    25