lhoangan / arap_flow Goto Github PK

Thanks for your wonderful code!

Will the generated optical flow data on DAVIS or QUVA be released? Hope to train PWC-net.

• enrivonment
  • ubuntu 16.04
  • CUDA 10.1 / 9.2
  • clang 3.8 / 9.0
  • NVIDIA V100

when i run arap_deform with provided sample, i faced below assertion problem.
Is there anything that i can try to solve problem?

./arap_deform cat512_iRGB.png cat512_iMsk.png cat512_iCstr.txt outFlow.flo outRGB.png outMask.png
Optimization plan at ../../arap_plan.t
warning: Linking two modules of different data layouts: '/usr/local/cuda/nvvm/libdevice/libdevice.compute_30.10.bc' is '' whereas 'external' is 'e-m:e-i64:64-f80:128-n8:16:32:64-S128'

nUnknowns = 786432
nResiduals = 0 + 262144 * 10

nnz = 0 + 262144 * 26

/arap-flow/ARAP/API/src/util.t:873: cuModuleLoadData: cuda reported error 218: a PTX JIT compilation failed
stack traceback:
[C]: in function 'error'
[string ""]:250: in function 'cudacompile'
arap-flow/ARAP/API/src/util.t:873: in function 'makeGPUFunctions'
.../arap-flow/ARAP/API/src/solverGPUGaussNewton.t:751: in function 'compilePlan'
arap-flow/ARAP/API/src/o.t:870: in function <arap-flow/ARAP/API/src/o.t:862>
[C]: in function 'xpcall'
arap-flow/ARAP/API/src/o.t:862: in function <arap-flow/ARAP/API/src/o.t:861>
arap_deform: src/../../shared/OptSolver.h:56: OptSolver::OptSolver(const std::vector &, const std::string &, const std::string &, bool): Assertion `m_plan' failed.
Aborted (core dumped)

i tried to run para_gen.py to create dataset with below arguments.

python3 para_gen.py --multseg --input /raid/hsson/datasets/DAVIS-480p --output /datasets/DMO/fd1 --fd 1 --img-pattern /datasets/FingerCamera

but there are some problems.

1. for first some images deformed and changed background works for input (frame1) and output(frame2)
   but after that, deformation and changing background only applied to input (frame1), not for output (frame2).