Пример #1
0
def get_image_classifier():
    """Load the CBCT HU slice classifier model. If the classifier is not locally available it will be downloaded."""
    classifier_file = osp.join(osp.dirname(__file__), 'demo_files',
                               'singleimage_classifier.pkl.gz')
    # get the classifier if it's not downloaded
    if not osp.isfile(classifier_file):
        logger.info("Downloading classifier from the internet...")
        classifier_file = retrieve_demo_file('singleimage_classifier.pkl.gz')
        logger.info("Done downloading")

    with gzip.open(classifier_file, mode='rb') as m:
        clf = pickle.load(m)
    return clf
Пример #2
0
def get_image_classifier(img_type):
    """Load the CBCT HU slice classifier model. If the classifier is not locally available it will be downloaded."""
    if img_type == 'single':
        classifier = 'singleimage_classifier.pkl.gz'
    elif img_type == 'vmat':
        classifier = 'vmat_classifier.pkl.gz'
    classifier_file = osp.join(osp.dirname(__file__), 'demo_files', classifier)
    # get the classifier if it's not downloaded
    if not osp.isfile(classifier_file):
        logger.info("Downloading classifier from the internet...")
        classifier_file = retrieve_demo_file(classifier)
        logger.info("Done downloading")

    with gzip.open(classifier_file, mode='rb') as m:
        clf = pickle.load(m)
    return clf
Пример #3
0
 def from_demo_image(cls):
     """Load the demo image into an instance"""
     demo_file = retrieve_demo_file(url='flatsym_demo.dcm')
     return cls(demo_file)
Пример #4
0
 def test_demo_is_reachable(self):
     if has_www_connection():
         file = retrieve_demo_file(url='flatsym_demo.dcm')
         self.assertTrue(osp.isfile(file))
Пример #5
0
 def get_filename(cls):
     return retrieve_demo_file(url='flatsym_demo.dcm')
Пример #6
0
 def test_demo_is_reachable(self):
     if has_www_connection():
         file = retrieve_demo_file(url=self.demo_name)
         self.assertTrue(osp.isfile(file))
Пример #7
0
 def from_demo_image(cls):
     """Instantiate and load the demo image."""
     demo_file = retrieve_demo_file(url=cls._demo_filename)
     return cls(demo_file)