1.详细统计数据集各个类别大中小目标数量(目标尺寸按照COCO数据集定义);
2.图像尺寸根据数据集中图像大小实时调整;
3.只需要修改图像路径和标签路径,即可运行。
COCO中大中小目标定义请查看论文与数据集:
import os
from pathlib import Path
import matplotlib.pyplot as plt
from PIL import Image
def getGtAreaAndRatio(label_dir, img_dir):
"""
得到不同尺度的gt框个数,并根据图像实际尺寸动态调整
:param label_dir: 标签文件夹路径
:param img_dir: 图像文件夹路径
:return: data_dict: {类别: {'area': [], 'ratio': []}}
"""
data_d