clip_visual_feats.py

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import os

import visual_extractor
import basic_utils as utils
from params import parse_args


def custom_path2id(path):
    return int(os.path.basename(path).split('.')[0].split('_')[-1])


def main(args):
    # load image list
    imgs = utils.read_lines(args.image_list)

    # build src_list & dst_list
    src_list = [os.path.join(args.image_dir, img) for img in imgs]
    dst_list = [os.path.join(args.output_dir, str(custom_path2id(img))+'.npz') for img in imgs]
    src_list, dst_list = utils.shuffle_list(src_list, dst_list)

    # extract
    worker = visual_extractor.create(args.ve_name, args, src_list, dst_list)
    worker.extract()


if __name__ == "__main__":
    args = parse_args()
    if not args.debug:
        utils.mkdirp(args.output_dir)
    main(args)