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