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);
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();