def main(): charts_cfg = cfg.clone() tables_cfg = cfg.clone() charts_cfg.merge_from_file(charts_path) tables_cfg.merge_from_file(tables_path) model_charts = ChataDemo(charts_cfg, "charts") model_tables = ChataDemo(tables_cfg, "tables") client = ApiClient() client.login() newest_ids = client.get_new_publications() for pub_id in newest_ids: pages = client.get_pages(pub_id) print(pub_id) for id, url in pages: print(id) print(url) resp = urllib.request.urlopen(url) image = np.asarray(bytearray(resp.read()), dtype="uint8") image = cv2.imdecode(image, cv2.IMREAD_COLOR) predictions_charts = model_charts.get_predictions(image) predictions_tables = model_tables.get_predictions(image) annotations_charts = annotate(predictions_charts) annotations_tables = annotate(predictions_tables) annotations = annotations_charts+annotations_tables print(annotations) if annotations: client.add_annotation(id, annotations)
def main(): if len(sys.argv) < 3: print( 'Usage: Path to model config path to model last_checkpoint model type' ) output_dir = sys.argv[2] model_type = sys.argv[3] cfg.merge_from_file(sys.argv[1]) cfg.merge_from_list(["OUTPUT_DIR", output_dir]) model = ChataDemo(cfg, model_type) client = ApiClient() client.login() #latest_id = client.get_latest() newest_ids = client.get_new_publications() for pub_id in newest_ids: pages = client.get_pages(pub_id) print(pub_id) for id, url in pages: print(id) print(url) resp = urllib.request.urlopen(url) image = np.asarray(bytearray(resp.read()), dtype="uint8") image = cv2.imdecode(image, cv2.IMREAD_COLOR) predictions = model.get_predictions(image) annotations = annotate(predictions) print(annotations) if annotations: print('writing annotation...') for annotation in annotations: client.add_annotation(id, annotation)