Open xsacha opened 5 years ago
These two test images look better in other implementations of RetinaFace for PyTorch, for eg. https://github.com/bogireddytejareddy/retinaface-pytorch/blob/master/test_results/t1.jpg https://github.com/bogireddytejareddy/retinaface-pytorch/blob/master/test_results/t4.jpg
These two test images look better in other implementations of RetinaFace for PyTorch, for eg. https://github.com/bogireddytejareddy/retinaface-pytorch/blob/master/test_results/t1.jpg https://github.com/bogireddytejareddy/retinaface-pytorch/blob/master/test_results/t4.jpg
Thanks for your test.And i noticed the problem as well , i am still trying to figure it out and try to make the performance and recall for rotated faces better. And i think there is some probable cause for this:
Landmarks are more accurate after your latest bug fix! Getting high precision (over 90%) quite early now at about epoch 10. Rotated faces are still not detected but I suppose that is due to the data loader.
Comparing to your result in README, which I was unable to achieve before:
---- [Epoch 39/200, Batch 400/403] ----
+----------------+-----------------------+
| loss name | value |
+----------------+-----------------------+
| total_loss | 0.09969855844974518 |
| classification | 0.09288528561592102 |
| bbox | 0.0034053439740091562 |
| landmarks | 0.003407923271879554 |
+----------------+-----------------------+
-------- RetinaFace Pytorch --------
Evaluating epoch 39
Recall: 0.7432201780921814
Precision: 0.906913273261629
I now get quite a bit better results than what you had: Recall of 75% and precision of 94.5%
---- [Epoch 39/200, Batch 1600/1610] ----
+----------------+-----------------------+
| loss name | value |
+----------------+-----------------------+
| total_loss | 0.08609515428543091 |
| classification | 0.0003947564400732517 |
| bbox | 0.035630300641059875 |
| landmarks | 0.05007009953260422 |
+----------------+-----------------------+
-------- RetinaFace Pytorch --------
Evaluating epoch 39
Recall: 0.7492150513297574
Precision: 0.9453832716431287
Just a note that epoch 39 was actually the best epoch I had out of the lot, so it only got worse from there.
Just a note that epoch 39 was actually the best epoch I had out of the lot, so it only got worse from there.
yes,I also noticed this.I am stilling trying to fix all these problems.
These two test images look better in other implementations of RetinaFace for PyTorch, for eg. https://github.com/bogireddytejareddy/retinaface-pytorch/blob/master/test_results/t1.jpg https://github.com/bogireddytejareddy/retinaface-pytorch/blob/master/test_results/t4.jpg