'/media/lenovo/zlf_ubuntu_backup/data/city/tfl_tsr/tsr-turnandright-0224-tr8val2/original/dataset/val/val.json',
]
annoFiles = train_annoFiles
catNms = generate_tfl_categories(abbr=False, type=1)
catNameNumsAll = {}
for name in catNms:
catNameNumsAll[name] = 0
totalNum = 0
for annoFile in annoFiles:
cocoGt = COCO(annoFile)
catIds = cocoGt.getCatIds(catNms)
_, catNameNums = cocoGt.countCatNums(catIds)
for name, num in catNameNums.items():
totalNum += num
if name in catNameNumsAll:
catNameNumsAll[name] += num
key_sorted_count = sorted(catNameNumsAll.items(), key=lambda x: x[0])
value_sorted_count = sorted(catNameNumsAll.items(),
key=lambda x: x[1],
reverse=True)
print("######## Category annotation numbers #######")
print("|{:<20}| {:<10}|".format('category', 'Numbers'))
print("|---------------|-------------------------------|")
print("|{:<20}| {:<10}|".format('total', totalNum))
for name, num in value_sorted_count:
