pythonlessons / TensorFlow-2.x-YOLOv3

YOLOv3 implementation in TensorFlow 2.3.1
https://pylessons.com/
MIT License
607 stars 327 forks source link

nan value in evaluate_mAP.py #188

Open hukutonpl opened 2 years ago

hukutonpl commented 2 years ago
t1 = time.time()
if YOLO_FRAMEWORK == "tf":
    pred_bbox = Yolo.predict(image_data)

predict gives nan value after the first image. This gives very low mAP

hukutonpl commented 2 years ago

Change it to this

t1 = time.time()
if YOLO_FRAMEWORK == "tf":
        Yolo.load_weights(f"./checkpoints/{TRAIN_MODEL_NAME}")
        pred_bbox = Yolo.predict(image_data)