| 123456789101112131415161718 |
- import argparse
- from .APMRToolkits import *
- dbName = 'human'
- def compute_APMR(dt_path, gt_path, target_key=None, mode=0):
- database = Database(gt_path, dt_path, target_key, None, mode)
- database.compare()
- mAP,_ = database.eval_AP()
- mMR,_ = database.eval_MR()
- line = 'AP:{:.4f}, MR:{:.4f}.'.format(mAP, mMR)
- return mAP, mMR
- if __name__ == "__main__":
- parser = argparse.ArgumentParser(description='Analyze a json result file with iou match')
- parser.add_argument('--detfile', required=True, help='path of json result file to load')
- parser.add_argument('--target_key', default=None, required=True)
- args = parser.parse_args()
- compute_APMR(args.detfile, args.target_key, 0)
|
