theAIGuysCode / yolov3_deepsort

Object tracking implemented with YOLOv3, Deep Sort and Tensorflow.
GNU General Public License v3.0
337 stars 182 forks source link

YoloV4 #24

Open LorBordin opened 4 years ago

LorBordin commented 4 years ago

Hi I'd like to use your script with a YoloV4, but load_weights.py fails in converting the model in tf format. Do you have any suggestion on how to make the script work for the YoloV4 as well?

its-charan-here commented 4 years ago

@LorBordin you won't be able to use the same script for running it on Yolov4 because the layers which are used in yolov3 are different than yolov4. the load_weights.py also loads the model from yolov3_tf2/models.py which is different, hence it is not possible to load yolov4 weights in this script