最近VisDrone19开放测试服务器了,自己尝试在它的test的图片上给出标签然后提交到官网上去。
使用了mmdetection工具,这个工具出来的结果是coco(json)格式的。
我们需要把它转化成官网需要的visdrone的txt格式才可以提交到服务器上,查看自己的score。
这个转化脚本目前找不到,所以只能自己写了。
import json
import os
import argparse
# wangzhifneg write in 2021-6-30
def coco2vistxt(json_path, out_folder):
labels = json.load(open(json_path, 'r', encoding='utf-8'))
for i in range(len(labels)):
# print(len(labels)) 158000
# print(labels[i]['image_id'])
# print(labels[i]['bbox'])
file_name = labels[i]['image_id'] + '.txt'
with open(os.path.join(args.o, file_name), 'a+', encoding='utf-8') as f:
l = labels[i]['bbox']
s = [round(i) for i in l]
line =str(s)[1:-1].replace(' ',''