Closed Markmichelin closed 3 years ago
Put an image in predict()
. Then, you can get the bbox.
Sorry i want to run in v2.1.0 what can i do ?
https://wiki.loliot.net/docs/lang/python/libraries/yolov4/python-yolov4-about#tensorflow
from yolov4.tf import YOLOv4
yolo = YOLOv4()
# yolo = YOLOv4(tiny=True)
yolo.classes = "coco.names"
yolo.input_size = (640, 480)
yolo.make_model()
yolo.load_weights("yolov4.weights", weights_type="yolo")
# yolo.load_weights("yolov4-tiny.weights", weights_type="yolo")
frame = cv2.imread(media_path)
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
bboxes = self.predict(frame_rgb)
Sorry i want to know How to get the coordinates of the bounding box in YOLOv4? thanks