def main(args):
dbf_parameters_dict = parse_parameters_json(args.dbf_parameters_json)
pprint.pprint(dbf_parameters_dict)
validation_detectors_detections = read_detectors_detections(
dbf_parameters_dict['validation_detections_filenames'])
annotations_dir = os.path.join(args.validation_dataset_dir, 'Annotations/')
images_dir = os.path.join(args.validation_dataset_dir, 'JPEGImages/')
validation_imagenames = read_imagenames(
args.validation_imagenames_filename, images_dir)
validation_annotations = read_annotations(
args.validation_pickled_annots_filename, annotations_dir,
validation_imagenames, args.dataset_name)
annotations_dir = os.path.join(args.test_dataset_dir, 'Annotations/')
images_dir = os.path.join(args.test_dataset_dir, 'JPEGImages/')
test_imagenames = read_imagenames(args.test_imagenames_filename,
images_dir)
test_annotations = read_annotations(args.test_pickled_annots_filename,
annotations_dir, test_imagenames,
args.dataset_name)
test_detectors_detections = read_detectors_detections(
dbf_parameters_dict['test_detections_filenames'])
validate_DBF(args.dataset_name, args.validation_dataset_dir,
validation_imagenames, validation_annotations,
validation_detectors_detections, args.test_dataset_dir,
test_imagenames, test_annotations, test_detectors_detections,
dbf_parameters_dict, args.map_iou_threshold,
args.output_filename)
def main(args):
detectors_detections = read_detectors_detections([args.detections_filename])
annotations_dir = os.path.join(args.dataset_dir, 'Annotations/')
images_dir = os.path.join(args.dataset_dir, 'JPEGImages/')
imagenames = read_imagenames(args.imagenames_filename, images_dir)
annotations = read_annotations(args.pickled_annots_filename, annotations_dir, imagenames, args.dataset_name)
cross_validate_base_detector(args.dataset_name, args.dataset_dir, imagenames,
annotations, detectors_detections, args.map_iou_threshold, args.folds_count)
def main(args):
annotations_dir = os.path.join(args.dataset_dir, 'Annotations/')
images_dir = os.path.join(args.dataset_dir, 'JPEGImages/')
imagenames = read_imagenames(args.imagenames_filename, images_dir)
annotations = read_annotations(args.pickled_annots_filename, annotations_dir, imagenames, args.dataset_name)
dbf_parameters_dict = parse_parameters_json(args.dbf_parameters_json)
pprint.pprint(dbf_parameters_dict)
detectors_detections = read_detectors_detections(dbf_parameters_dict['validation_detections_filenames'])
cross_validate_ALFA(args.dataset_name, args.dataset_dir, imagenames, annotations, detectors_detections,
dbf_parameters_dict, args.map_iou_threshold, args.folds_count)
def main(args):
nms_parameters_dict = parse_parameters_json(args.nms_parameters_json)
pprint.pprint(nms_parameters_dict)
detectors_detections = read_detectors_detections(
nms_parameters_dict['detections_filenames'])
annotations_dir = os.path.join(args.dataset_dir, 'Annotations/')
images_dir = os.path.join(args.dataset_dir, 'JPEGImages/')
imagenames = read_imagenames(args.imagenames_filename, images_dir)
annotations = read_annotations(args.pickled_annots_filename,
annotations_dir, imagenames,
args.dataset_name)
validate_NMS(args.dataset_name, args.dataset_dir, imagenames, annotations,
detectors_detections, nms_parameters_dict,
args.map_iou_threshold, args.output_filename)
def main(args):
if os.path.isfile(args.detections_filename):
with open(args.detections_filename, 'rb') as f:
if sys.version_info[0] == 3:
images_detections = pickle.load(f, encoding='latin1')
else:
images_detections = pickle.load(f)
else:
print('Detections filename was not found!')
exit(1)
annotations_dir = os.path.join(args.dataset_dir, 'Annotations/')
images_dir = os.path.join(args.dataset_dir, 'JPEGImages/')
imagenames = read_imagenames(args.imagenames_filename, images_dir)
annotations = read_annotations(args.pickled_annots_filename,
annotations_dir, imagenames,
args.dataset_name)
map_computation = Computation_mAP(None)
map_computation.compute_map(args.dataset_name, args.dataset_dir,
imagenames, annotations, images_detections,
args.map_iou_threshold)