candidates.py 文件源码

def __init__(self, config):
    super(CandidateDataLoader, self).__init__(config)

    self.resize_to = config.size[0]
    self.diameter_mm = config.diameter_mm
    self.batch_size = config.batch
    self.validation_rate =config.validation_ratio
    self.k = config.top_k
    # set data dir and file
    self.data_dir = 'data/'
    self.pkl_dir = self.data_dir + 'pkl/'
    self.stage1_dir = self.data_dir + 'stage1/'
    self.sample_dir = self.data_dir + 'sample/'
    self.patch_dir = self.data_dir + 'patch/'
    self.data_file_name = 'kaggle_stage1.pkl'
    self.feature_file_name = 'features_detect_kaggle.pkl'

    self.train_ids, self.valid_ids, self.test_ids = self.split_dataset()
    # self.train_ids, self.valid_ids, self.test_ids  = self.get_ids_from_sample_dataset()
    self.current_ids = self.train_ids
    self.current_pointer = 0
    self.current_set = None
    self.random_for_negative_samples = True

    self.data = self.build_data_dict(layer_features=['67', '77'], k=self.k)