Exemplo n.º 1
0
def main(argv):      
  from CmdArgumentsHelper import CmdArgumentsHelper;
  arg_helper = CmdArgumentsHelper();
  arg_helper.add_argument('query', 'q', 'query', 1);
  arg_helper.add_argument('root_dir', 'r', 'root', 1);
  arg_helper.add_argument('output_dir', 'o', 'outdir', 1);
  args = arg_helper.read_arguments(argv);
  print (args);

  query = args['query'];
  images = gen_flickr_image_info(args['root'], query);
  image_urls = images['image_urls'];
  image_ids = images['image_ids'];
  downloader = ImageDownloader();
  output_dir = args['output_dir'] + query;
  from FileIO import FileIO;
  fileIO = FileIO();
  fileIO.create_folders(output_dir);
  downloader.download_images(image_urls, image_ids, output_dir);
Exemplo n.º 2
0
def main():
    arg_helper = CmdArgumentsHelper();
    arg_helper.add_argument('query', 'q', 'query', 1);
    arg_helper.add_argument('root_dir', 'r', 'root', 1);
    arg_helper.add_argument('top_tag_num', 't', 'top_tag_num', 1);    
    args = arg_helper.read_arguments();
    print (args);

    root = args['root_dir'];
    top_tag_num = int(args['top_tag_num']);
    query = args['query'];

    from PhotoDao import PhotoDao;
    from DBHelper import DBHelper;
    db_helper = DBHelper();
    db_helper.init(root);
    photo_dao = PhotoDao(db_helper);

    print('get photo ids.');
    tic();
    photo_ids = photo_dao.getClassPhotoIds(query, ''.join([query]));
    toc();

    print('get photos ...');
    tic();
    photos = photo_dao.getPhotos(query, photo_ids);
    toc();

    print('gen top tags via user...');
    from task_gen_top_tags import task_gen_top_tag_via_user;
    tic();
    top_word = task_gen_top_tag_via_user(photos, query, top_tag_num, root);
    toc();

    print('build tag features...');
    from task_gen_top_tags import task_build_tag_features;
    tic();
    task_build_tag_features(top_word, query, photos, 0, root);
    toc();