Updated so .h5 files can be loaded in to make predictions

taipingeric / yolo-v4-tf.keras

A simple tf.keras implementation of YOLO v4

MIT License

140 stars 79 forks source link

Open BrentZ-1849203 opened 3 years ago

BrentZ-1849203 commented 3 years ago

Added .h5 check to if statement so code to load in .h5 files can be reached.

sadransh commented 2 years ago

A question on stackoverflow related to this: https://stackoverflow.com/a/71696396/1076264