YOLOv1-TensorFlow2.0 icon indicating copy to clipboard operation
YOLOv1-TensorFlow2.0 copied to clipboard

A tensorflow2.0 implementation of the YOLOv1 paper https://arxiv.org/pdf/1506.02640

Results 2 YOLOv1-TensorFlow2.0 issues
Sort by recently updated
recently updated
newest added

In line 200 def load_data(image_path, labels): flip = tf.cast(tf.random.uniform(shape[1,],minval=0,maxval=2,dtype=tf.int32),dtype=tf.bool).numpy()[0] return get_image(image_path), flip_labels(labels)#don't define flip_labels()?

Yolo loss is above 35 after 30 epochs. Need help