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.98k stars 524 forks source link

I did not get any output #51

Open MangoloD opened 2 years ago

MangoloD commented 2 years ago

I have ran detect.py with yolor_p6.pt but didn't get any output image

MangoloD commented 2 years ago

image I find that the pred result is nan

WongKinYiu commented 2 years ago

what is your command for making prediction?

MangoloD commented 2 years ago

I have solved the problem,just set 'half=False' but I don't know why @WongKinYiu

Orangeman69 commented 2 years ago

Same problem.

FXY0117 commented 2 years ago

我已经解决了这个问题,只是设置了 'half=False' 但我不知道为什么@WongKinYiu I meet this question too, where is your this command?

WongKinYiu commented 2 years ago

https://github.com/WongKinYiu/yolor/blob/main/detect.py#L40