yolov5检测限定长宽比检测范围的目标

yolov5检测限定长宽比检测范围的目标

正常环境下，我们所使用的数据中，不同类别的目标尺寸范围都有一定的范围，在工业缺陷检测领域尤其如此，因此我们修改yolov5使其能针对特定目标只输出特定区域的目标。 如： A目标长宽比大于10小于20， B目标长宽比大于20小于40等 修改如下：

train.py

parser.add_argument('--ar_thr', nargs='+', type=int, default=[101,130,46,12,17], help='ar_thr of dataset to be used')

datasets.py

def box_candidates(lbl, box1, box2, wh_thr=2, ar_thr=20, area_thr=0.1, eps=1e-16): # box1(4,n), box2(4,n) # Compute candidate boxes: box1 before augment, box2 after augment, wh_thr (pixels), aspect_ratio_thr, area_ratio w1, h1 = box1[2] - box1[0], box1[3] - box1[1] w2, h2 = box2[2] - box2[0], box2[3] - box2[1] ar = np.maximum(w2 / (h2 + eps), h2 / (w2 + eps)) # aspect ratio # return (w2 > wh_thr) & (h2 > wh_thr) & (w2 * h2 / (w1 * h1 + eps) > area_thr) & (ar < ar_thr) # candidates ar_thr_flag = np.array([ar0 < ar_thr[int(lbl0)] for lbl0, ar0 in zip(lbl, ar)]) return (w2 > wh_thr) & (h2 > wh_thr) & (w2 * h2 / (w1 * h1 + eps) > area_thr) & ar_thr_flag # candidates

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。