WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

keep aspect ratio

MasIgor opened this issue · comments

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