import pickle
import json
import argparse
import mmcv

def parse_args():
    parser = argparse.ArgumentParser(
        description='converting results to json format')
    parser.add_argument('rname')
    args = parser.parse_args()
    return args


if __name__ == '__main__':
    args = parse_args()
    rname = args.rname
    r = pickle.load(open(rname, 'rb'))

    from mmdet.datasets import build_dataloader, build_dataset
    cfg = mmcv.Config.fromfile('/data/dibe/multi-phase/configs/multiphase/coco/faster_rcnn_r50_fpn_1x_coco_pl.pyy')
    cfg.data.test.test_mode = True
    dataset = build_dataset(cfg.data.test)

    dataset.results2json(r, rname.split('.')[0])