YOLOP
YOLOP copied to clipboard
fixed plot dimension error
The demo.py file has an error plotting the results, which is fixed here.
This fix solves the dimension problem and allows us to infer images on any shapes. Thanks @amirhosseinh77