search

visinf / irr

Iterative Residual Refinement for Joint Optical Flow and Occlusion Estimation (CVPR 2019)
Apache License 2.0
194 stars 32 forks source link

Args Module #26

Closed AliKafaei closed 4 years ago

AliKafaei commented 4 years ago

Thanks for your complete code. I want to use the network in my own code (not using in console command mode) so I need to load state dictionary and pass it to the network. The problem is that the network model needs an input argument args. Most of the args parameters can be ignored except args.modules(). After reading the code for several times; I couldent find any clue that what it is and how can I set it manually. Selection_001

What should I set self.modules()? Another issue is that scipy no longer supports imread and it must be replaced by cv2,imread. Best, Ali Kafaei

hurjunhwa commented 4 years ago

Hi,

In the original code, I used tqdm with a version of 4.24.0. But using up to 4.40.0 will still work.

Regarding the self.module(), probably referring this will help: https://discuss.pytorch.org/t/pytorch-self-module/49677/5

Yes, unfortunately scipy no longer support. Using cv2.imread or skimage.io.imread can be the good alternative as you mention.

Thanks, Jun