seg_dataset_assembler.py 文件源码

def __init__(self, height, width, output_fname, semantic_only=True):
    self.height = height
    self.width = width
    self.semantic_only = semantic_only
    self.log = logger.get()
    self.output_fname = output_fname

    self.log.info("Output h5 dataset: {}".format(self.output_fname))
    self.log.info("Reading image IDs")
    self.img_ids = self.read_ids()

    # Shuffle sequence.
    random = np.random.RandomState(2)
    shuffle = np.arange(len(self.img_ids))
    random.shuffle(shuffle)
    self.img_ids = [
        self.img_ids[shuffle[idx]] for idx in range(len(self.img_ids))
    ]
    pass