修改pycocotools输出每个类别AP结果

参考：https://github.com/tloki/cocoapi/commit/52c206187e3415b60318f8b05e7dc405b6e8514e

 

            print(
                "category names: {}".format([e["name"] for e in sorted(dataset["categories"], key=lambda x: x["id"])]))

 

            category_dimension = 1 + int(ap)
            if s.shape[category_dimension] > 1:

                iStr += ", per category = {}"
                mean_axis = (0,)
                if ap == 1:
                    mean_axis = (0, 1)
                per_category_mean_s = np.mean(s, axis=mean_axis).flatten()
                with np.printoptions(precision=3, suppress=True, sign=" ", floatmode="fixed"):
                    print(iStr.format(titleStr, typeStr, iouStr, areaRng, maxDets, mean_s, per_category_mean_s))
            else:
                print(iStr.format(titleStr, typeStr, iouStr, areaRng, maxDets, mean_s, ""))