mirrors/mmselfsup
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmselfsup.git synced 2025-06-03 14:59:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Fix] fix dataset.evaluate args bug (#192)

Browse Source
This commit is contained in:
Yixiao Fang 2022-01-28 16:46:53 +08:00 committed by GitHub
parent d029a0db05
commit c39bb83a7c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/test.py
View File

@ -18,7 +18,7 @@ from mmselfsup.utils import (get_root_logger, multi_gpu_test,
def parse_args(): def parse_args():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description='MMDet test (and eval) a model') description='MMSelfSup test (and eval) a model')
parser.add_argument('config', help='test config file path') parser.add_argument('config', help='test config file path')
parser.add_argument('checkpoint', help='checkpoint file') parser.add_argument('checkpoint', help='checkpoint file')
parser.add_argument( parser.add_argument(
@ -86,6 +86,9 @@ def main():
distributed = True distributed = True
init_dist(args.launcher, **cfg.dist_params) init_dist(args.launcher, **cfg.dist_params)
# create work_dir
mmcv.mkdir_or_exist(osp.abspath(cfg.work_dir))
# logger # logger
timestamp = time.strftime('%Y%m%d_%H%M%S', time.localtime()) timestamp = time.strftime('%Y%m%d_%H%M%S', time.localtime())
log_file = osp.join(cfg.work_dir, f'test_{timestamp}.log') log_file = osp.join(cfg.work_dir, f'test_{timestamp}.log')
@ -116,8 +119,7 @@ def main():
rank, _ = get_dist_info() rank, _ = get_dist_info()
if rank == 0: if rank == 0:
for name, val in outputs.items(): dataset.evaluate(outputs, logger, topk=(1, 5))
dataset.evaluate(torch.from_numpy(val), name, logger, topk=(1, 5))
if __name__ == '__main__': if __name__ == '__main__':