def build_coco_dataset():
# Build the database
raw_db, _ = build_singleobj_database()
patch_db, _ = build_patch_database(raw_db, 1000)
# Build and formatter and run it
formatter_config = COCODatasetFormatterConfig()
formatter_config.db_name = 'shoe_test'
formatter_config.base_folder = '/home/wei/data/pdc/coco/shoe_test'
formatter = COCODatasetFormatter(formatter_config)
formatter.process_db_list([patch_db])
def build_stick_dataset_full():
# Build the database
raw_db, _ = make_singleobj_database("/pdc/logs_proto/stick_samples_2021-01-30_all.txt", "stick")
patch_db, _ = build_patch_database([raw_db], 70000)
# Build and formatter and run it
formatter_config = COCODatasetFormatterConfig()
formatter_config.db_name = "stick_db"
formatter_config.base_folder = "/pdc/logs_proto/coco/stick_db"
formatter = COCODatasetFormatter(formatter_config)
formatter.process_db_list([patch_db])
def build_coco_dataset():
# Build the database
raw_db, _ = make_singleobj_database(
'/home/wei/Coding/mrcnn_integrate/dataset_config/wiper.txt', 'wiper')
patch_db, _ = build_patch_database([raw_db], 20000)
# Build and formatter and run it
formatter_config = COCODatasetFormatterConfig()
formatter_config.db_name = 'wiper_db'
formatter_config.base_folder = '/home/wei/data/coco/wiper_db'
formatter = COCODatasetFormatter(formatter_config)
formatter.process_db_list([patch_db])
def build_coco_dataset():
# Build the database
raw_db_peg, _ = make_singleobj_database('/home/wei/Coding/mrcnn_integrate/dataset_config/siemens_peg.txt', 'peg')
raw_db_hole, _ = make_singleobj_database('/home/wei/Coding/mrcnn_integrate/dataset_config/siemens_hole.txt', 'hole')
patch_db, _ = build_patch_database([raw_db_peg, raw_db_hole], 20000)
# Build and formatter and run it
formatter_config = COCODatasetFormatterConfig()
formatter_config.db_name = 'peghole_db'
formatter_config.base_folder = '/home/wei/data/coco/peghole_db'
formatter = COCODatasetFormatter(formatter_config)
formatter.process_db_list([patch_db])
def build_coco_dataset():
# Build the database
raw_db, _ = make_singleobj_database(
'/home/wei/Code/mankey_ros/mankey/config/spartan_data/mugs_all.txt',
'mug')
patch_db, _ = build_patch_database([raw_db], 400)
# Build and formatter and run it
formatter_config = COCODatasetFormatterConfig()
formatter_config.db_name = 'mug_db'
formatter_config.base_folder = '/home/wei/data/coco/mug_db'
formatter = COCODatasetFormatter(formatter_config)
formatter.process_db_list([patch_db])