Yolov3_medium_4

detect.py

def main():
    # load our input image and grab its spatial dimensions
    image = cv2.imread("./test1.jpg")
    labelsPath="yolo_v3/coco.names"
    cfgpath="yolo_v3/yolov3.cfg"
    wpath="yolo_v3/yolov3.weights"
    Lables=get_labels(labelsPath)
    CFG=get_config(cfgpath)
    Weights=get_weights(wpath)
    nets=load_model(CFG,Weights)
    Colors=get_colors(Lables)
    res=get_predection(image,nets,Lables,Colors)
    # image=cv2.cvtColor(image,cv2.COLOR_BGR2RGB)
    # show the output image
    cv2.imshow("Image", res)
    cv2.waitKey()