rigid segmentation

[daonguyenthien]

Hi,
can you released your code on the rigid segmentation? i seem to be stuck at this part
Thank you

[gengshan-y]

Hi, can you be more specific about which part of the code are your looking for?

[daonguyenthien]

Hi, I'm very sorry for not being clear.
I'm looking for the part on how you can segment motion from the background?
and how can you fusion it with your optical flow model?
Thank you.

[gengshan-y]

The reference command segments rigid motions and refines the scene flow.

Specifically, the rigid motion segmentation part starts here:

rigidmask/models/VCNplus.py

Line 729 in 52a8c0d

costs = torch.cat((

and the part that calls scene flow refinement part starts here:

rigidmask/submission.py

Line 325 in 52a8c0d

scene_type, T01_c, R01,RTs = ddlib.rb_fitting(bgmask,mask_input,disp,flow,occ,K0,K1,bl,parallax_th=4,mono=(args.sensor=='mono'), sintel='Sintel' in idxname)