示例#1
0
def read_images():
    galaxy_ids = get_galaxy_ids()
    for galaxy_id in galaxy_ids:
        image_file = '.'.join([str(galaxy_id), 'jpg'])
        feature_vector = [galaxy_id]
        features = Features(os.path.join(cfg.TRAIN_IMAGE_DIR, image_file))
        feature_vector.extend(features.get_feature_vector())
        yield feature_vector