HanLingsgjk / TPCV

code for CVPR23 Paper Learning optical expansion from scale matching
BSD 3-Clause "New" or "Revised" License
3 stars 0 forks source link

No module named 'core.raft' #2

Open Haalum opened 1 year ago

Haalum commented 1 year ago

Is the raft script missing in core? The error is from the following line in evaluation.py and demo.py. Thanks!

from core.raft import RAFT

I'm getting the following error.

ModuleNotFoundError: No module named 'core.raft'
HanLingsgjk commented 1 year ago

Because this code is based on KITTI modifications (the demo and evaluation are from the original RAFT), I did not release the demo, Currently, only training(train.py) and evaluation(dc_flow_eval.py) are open.

Haalum commented 1 year ago

Thanks!

I'm new to this. A quick question. I have downloaded the stereo flow KITTI data from here. Download stereo 2015/flow 2015/scene flow 2015 data set (2 GB)

I got the following folders in the downloaded training folder.

-testing
-training
   --disp_noc_0  
   --disp_noc_1  
   --disp_occ_0  
   --disp_occ_1  flow_noc  
   --flow_occ 
   -- image_2  
   --image_3  
   --obj_map  
   --viz_flow_occ  
   --viz_flow_occ_dilate_1

In dataset_exp_orin.py, you have the following line

masko = sorted(glob(osp.join(root, 'mask_img/*_10.png')))

I don't have mask_img folder in the downloaded data. Can you please point me to the right direction? Thanks.

HanLingsgjk commented 1 year ago

I'm very sorry for my negligence. This mask is of no use here. You can delete the parts related to it.