lulindeng

> Hi @with-twilight, it seems the problem is caused by the PyTorch environment. Our code is tested on PyTorch 1.2, so there may be some bugs with PyTorch1.0. Hi, thanks...

> @lulindeng I met with similar problems before. After I switch to use pytorvh==1.6.0, the problem disappears. Thank you ! I solved the problem by using revised code in this...

> The steps as follows can solve the problem: > > ``` > sudo apt update > sudo apt-get install llvm-6.0 > sudo apt-get install libsm6 libxrender1 libfontconfig1 freeglut3-dev >...

> @lulindeng you can change "rgb[:, :, :-1]" into "rgb[:, :, :]"，the error will disappear, good luck @xiezhongzhao It works, thanks a lot!