Skip to content
Snippets Groups Projects
Commit 768e716e authored by nikhil_rayaprolu's avatar nikhil_rayaprolu
Browse files

Update test.py

parent 7b569874
No related branches found
No related tags found
No related merge requests found
...@@ -598,7 +598,6 @@ def main(): ...@@ -598,7 +598,6 @@ def main():
coco_eval(result_files, eval_types, dataset.coco) coco_eval(result_files, eval_types, dataset.coco)
print(args.json_out, rank) print(args.json_out, rank)
if outputs and args.json_out and rank == 0: if outputs and args.json_out and rank == 0:
print(outputs)
if not isinstance(outputs[0], dict): if not isinstance(outputs[0], dict):
response = results2json(dataset, outputs, args.json_out) response = results2json(dataset, outputs, args.json_out)
else: else:
...@@ -606,7 +605,6 @@ def main(): ...@@ -606,7 +605,6 @@ def main():
outputs_ = [out[name] for out in outputs] outputs_ = [out[name] for out in outputs]
result_file = args.json_out + '.{}'.format(name) result_file = args.json_out + '.{}'.format(name)
response = results2json(dataset, outputs_, result_file) response = results2json(dataset, outputs_, result_file)
print(response, response['segm'], args.json_out)
shutil.move(response['segm'], predictions_output_path) shutil.move(response['segm'], predictions_output_path)
if __name__ == '__main__': if __name__ == '__main__':
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment