mystic123 / tensorflow-yolo-v3

Implementation of YOLO v3 object detector in Tensorflow (TF-Slim)
https://medium.com/@pawekapica_31302/implementing-yolo-v3-in-tensorflow-tf-slim-c3c55ff59dbe
Apache License 2.0
893 stars 353 forks source link

Get input image shape statically in the model construction function #101

Open LucasMahieu opened 3 years ago

LucasMahieu commented 3 years ago

Instead of using input.get_shape() fonction We can pass input image shape to yolo_v3 and yolo_v3_tiny constructor function. Since the input placeholder shape is statically defined (to None, size, size, 3) We can have access to this 'size' when constructing the yolo_v3 or yolo_v3_tiny models.

This if more efficient for inference. This commit should not break any anterior codes since it is only adding 1 optional argument to model constructor functions