WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.99k stars 518 forks source link

keep aspect ratio #255

Open MasIgor opened 2 years ago

MasIgor commented 2 years ago

Hello!

I have multiple similar objects, that have different aspect ratios. But for the rest they look pretty much the same.. I do get a lot of misinterpretations on those elements. Is there a way to tell yolor that during the training it should also keep track of the shape/proportion of the object? for isntance an ellypse is not a circle.. tagging them both in my dataset did not solve the issue. During training it keeps misdetecting them and the precision tops at about 0.78

thank you