search

gengshan-y / rigidmask

Code for "Learning to Segment Rigid Motions from Two Frames". CVPR 2021.
https://gengshan-y.github.io/rigidmask/
MIT License
191 stars 18 forks source link

problem #7

Open xubangwu opened 3 years ago

xubangwu commented 3 years ago

hello,I have a probelm, in the 'exploader.py' , the 168 line shows us that change_size[:,:,1:4]=d1,d2,d2 and change_size[:,:,4:6]=flow3d, then we get the rectified 3D flow named 'p3d' from the 202 line, but the 203 line shows that we replace the 'd1,d2,d2' with the rectified 3D flow named 'p3d', I have the problem why we do not replace the old flow3d which is change_size[:,:,4:6] with 'p3d'?

gengshan-y commented 3 years ago

Hi, I may not follow your question. 3d flows are not replaced because they will be used at training time. I updated exploader.py to make it clearer. The code is not cleaned up yet and sorry for any confusion due to variable names.

xubangwu commented 3 years ago

Thank you very much. I understand what you mean.