Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

About option "keep_aspect_ratio" in examples/object_detection.py #36

Closed
tongtybj opened this issue Dec 27, 2019 · 1 comment
Closed

Comments

@tongtybj
Copy link

I followed the instruction here, then I got the following result and first figure:

$ python3 examples/object_detection.py --model='test_data/ssd_mobilenet_v1_fine_tuned_edgetpu.tflite' --label='test_data/pet_labels.txt' --input='test_data/pets.jpg' --keep_aspect_ratio
-----------------------------------------
german_shorthaired
score =  0.37109375
box =  [1762.1801829338074, 838.6267263974463, 2210.349624156952, 1173.771092295647]
-----------------------------------------
shiba_inu
score =  0.32421875
box =  [1916.2815427780151, 784.8315710467953, 2195.608711242676, 1104.3338206197534]
Please check  object_detection_result.jpg

object_detection_result

I can see both two bounding boxes are assigned to the right dog.

Then I remove --keep_aspect_ratio form command line, which gave a better result.

$ python3 examples/object_detection.py --model='test_data/ssd_mobilenet_v1_fine_tuned_edgetpu.tflite' --label='test_data/pet_labels.txt' --input='test_data/pets.jpg' -----------------------------------------
staffordshire_bull_terrier
score =  0.37109375
box =  [712.9122591018677, 272.12397533655167, 1122.5919842720032, 674.8110462427139]
-----------------------------------------
staffordshire_bull_terrier
score =  0.35546875
box =  [1831.8864297866821, 770.1564584970474, 2190.160117149353, 1146.7327305078506]
Please check  object_detection_result.jpg

object_detection_result

@Namburger
Copy link

@tongtybj Thanks for submitting the issue.
I believe this is most likely just due to the input images that was feed to the model during training did not have paddings. You can check the option here and here

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants