最新国产好看的视频,伊人天堂AV在线,国产Aaaaaa视频,蜜臀视频在线观看一区,人妻av色图,密臀久久久精品影片,青青视频免费观看毛片,久草在线观看视,国产三级精品色情在线

如何使用yolov5輸出檢測(cè)到的目標(biāo)坐標(biāo)信息

 更新時(shí)間:2022年03月28日 09:34:45   作者:一位安分的碼農(nóng)  
YOLOv5是一系列在 COCO 數(shù)據(jù)集上預(yù)訓(xùn)練的對(duì)象檢測(cè)架構(gòu)和模型,下面這篇文章主要給大家介紹了關(guān)于如何使用yolov5輸出檢測(cè)到的目標(biāo)坐標(biāo)信息的相關(guān)資料,需要的朋友可以參考下

找到detect.py,在大概113行,找到plot_one_box

? ? ? ? ? ? ? ? # Write results
? ? ? ? ? ? ? ? for *xyxy, conf, cls in reversed(det):
? ? ? ? ? ? ? ? ? ? if save_txt: ?# Write to file
? ? ? ? ? ? ? ? ? ? ? ? xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(-1).tolist() ?# normalized xywh
? ? ? ? ? ? ? ? ? ? ? ? with open(txt_path + '.txt', 'a') as f:
? ? ? ? ? ? ? ? ? ? ? ? ? ? f.write(('%g ' * 5 + '\n') % (cls, *xywh)) ?# label format

? ? ? ? ? ? ? ? ? ? if save_img or view_img: ?# Add bbox to image
? ? ? ? ? ? ? ? ? ? ? ? label = '%s %.2f' % (names[int(cls)], conf)
? ? ? ? ? ? ? ? ? ? ? ? plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)

ctr+鼠標(biāo)點(diǎn)擊,進(jìn)入general.py,并自動(dòng)定位到plot_one_box函數(shù),修改函數(shù)為

def plot_one_box(x, img, color=None, label=None, line_thickness=None):
    # Plots one bounding box on image img
    tl = line_thickness or round(0.002 * (img.shape[0] + img.shape[1]) / 2) + 1  # line/font thickness
    color = color or [random.randint(0, 255) for _ in range(3)]
    c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))
    cv2.rectangle(img, c1, c2, color, thickness=tl, lineType=cv2.LINE_AA)
    print("左上點(diǎn)的坐標(biāo)為:(" + str(c1[0]) + "," + str(c1[1]) + "),右下點(diǎn)的坐標(biāo)為(" + str(c2[0]) + "," + str(c2[1]) + ")")

即可輸出目標(biāo)坐標(biāo)信息了

附:python yolov5檢測(cè)模型返回坐標(biāo)的方法實(shí)例代碼

python yolov5檢測(cè)模型返回坐標(biāo)的方法 直接搜索以下代碼替換下 

 if save_img or view_img:  # Add bbox to image
                        label = f'{names[int(cls)]} {conf:.2f}'
                        c1, c2 = (int(xyxy[0]), int(xyxy[1])), (int(xyxy[2]), int(xyxy[3]))
                        print("左上點(diǎn)的坐標(biāo)為:(" + str(c1[0]) + "," + str(c1[1]) + "),右下點(diǎn)的坐標(biāo)為(" + str(c2[0]) + "," + str(c2[1]) + ")")
                        return [c1,c2]
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pt', help='model.pt path(s)')
parser.add_argument('--source', type=str, default='data/images', help='source') # file/folder, 0 for webcam
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
parser.add_argument('--view-img', action='store_true', help='display results')
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
parser.add_argument('--nosave', action='store_true', help='do not save images/videos')
parser.add_argument('--classes', nargs='+', type=int, help='filter by class: --class 0, or --class 0 2 3')
parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS')
parser.add_argument('--augment', action='store_true', help='augmented inference')
parser.add_argument('--update', action='store_true', help='update all models')
parser.add_argument('--project', default='runs/detect', help='save results to project/name')
parser.add_argument('--name', default='exp', help='save results to project/name')
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
opt = parser.parse_args()

check_requirements(exclude=('pycocotools', 'thop'))

opt.source='data/images/1/'
result=detect()
print('最終檢測(cè)結(jié)果:',result);

總結(jié)

到此這篇關(guān)于如何使用yolov5輸出檢測(cè)到的目標(biāo)坐標(biāo)信息的文章就介紹到這了,更多相關(guān)yolov5輸出目標(biāo)坐標(biāo)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!

相關(guān)文章

最新評(píng)論

广宗县| 健康| 敖汉旗| 南宁市| 永清县| 福州市| 佛冈县| 兴安县| 临沧市| 英超| 合阳县| 涟水县| 屯留县| 安远县| 桃江县| 法库县| 青冈县| 南昌市| 页游| 内江市| 桐柏县| 墨竹工卡县| 双牌县| 铁力市| 南投市| 全南县| 彭州市| 河间市| 泸西县| 河北省| 灵璧县| 合山市| 高淳县| 禹州市| 宜州市| 兴国县| 乳山市| 宁化县| 黎川县| 察雅县| 岗巴县|