Python crop_and_preprocessの例

コード例 #1

ファイル: Captionizer.py プロジェクト: manishmeshram51/caption-generator

def apply_model_to_image_raw_bytes(img):
    # img = utils.decode_image_from_buf(raw)
    # fig = plt.figure(figsize=(7, 7))
    # plt.grid('off')
    # plt.axis('off')
    # plt.imshow(img)
    img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE), final_model.preprocess_for_model)
    # print(' '.join(generate_caption(img)[1:-1]))
    return str(' '.join(generate_caption(img)[1:-1]))

コード例 #2

ファイル: caption_image.py プロジェクト: chayan/image_captioning

def apply_model_to_image_raw_bytes(raw, sample=False):
    img = utils.decode_image_from_buf(raw)
    plt.figure(figsize=(7, 7))
    plt.grid('off')
    plt.axis('off')
    plt.imshow(img)
    img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE), final_model.preprocess_for_model)
    print(' '.join(generate_caption(img, sample=sample)[1:-1]))
    plt.show()

コード例 #3

ファイル: image_captioning.py プロジェクト: akirov/Coursera

def apply_model_to_image_raw_bytes(raw, fname=None, do_save=False):
    img = utils.decode_image_from_buf(raw)
    fig = plt.figure(figsize=(7, 7))
    plt.grid('off')
    plt.axis('off')
    plt.imshow(img)
    img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE),
                                    final_model.preprocess_for_model)
    plt.title(' '.join(generate_caption(img)[1:-1]))
    if do_save:
        plt.savefig(fname)
    plt.show()

コード例 #4

ファイル: demo.py プロジェクト: yiruijiang/deep_learning_projects

def apply_model_to_image_raw_bytes(raw):
    img = utils.decode_image_from_buf(raw)
    plt.figure(figsize=(7, 7))
    plt.grid('off')
    plt.axis('off')
    plt.imshow(img)
    img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE),
                                    generator_model.process_for_model)
    img = np.expand_dims(img, axis=0)
    title = generate_caption(img)[1:-1]
    plt.title(' '.join(title))
    plt.show()

コード例 #5

ファイル: demo.py プロジェクト: yiruijiang/deep_learning_projects

def show_valid_example_series(val_img_fns, index_array):
    all_files = set(val_img_fns)
    found_files = list(
        filter(lambda x: x.filename.rsplit("/")[-1] in all_files, zf.filelist))
    #plt.figure()
    index = 1
    for row in range(index_array.shape[0]):
        for col in range(index_array.shape[1]):
            fn = found_files[index_array[row, col]]
            img = utils.decode_image_from_buf(zf.read(fn))
            subplot = plt.subplot(index_array.shape[0], index_array.shape[1],
                                  index)
            subplot.axis('off')
            subplot.grid('off')
            plt.imshow(img)
            img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE),
                                            generator_model.process_for_model)
            img_s = np.expand_dims(img, axis=0)
            title = generate_caption(img_s, sample=False)[1:-1]
            subplot.set_title(" ".join(title))
            index += 1
    plt.show()

コード例 #6

def apply_model_to_image_raw_bytes(raw):
    img = utils.decode_image_from_buf(raw)
    img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE),
                                    final_model.preprocess_for_model)
    print(' '.join(generate_caption(img)[1:-1]))

コード例 #7

def apply_model_to_image_raw_bytes_return_next_word_probs(raw, par_caption):
    img = utils.decode_image_from_buf(raw)
    img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE),
                                    final_model.preprocess_for_model)
    return generate_next_word_probs(img, par_caption)

コード例 #8

def apply_model_to_image(fname):
    img = imread(fname)[:,:,:3]
    img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE), final_model.preprocess_for_model)
    return (' '.join(generate_caption(img)[1:-1]))

コード例 #9

ファイル: demo.py プロジェクト: yiruijiang/deep_learning_projects

def get_caption(img):
    img = utils.crop_and_preprocess(img, (IMG_SIZE, IMG_SIZE),
                                    generator_model.process_for_model)
    img = np.expand_dims(img, axis=0)
    return generate_caption(img)[1:-1]