evaluate.py

[KirillHiddleston]

Hi nice work.
Coud u tell me pls. Whats wrong with this, after run python evaluate.py I have error.

python evaluate.py --data_path /Users/kgkozlov/Downloads/img_4.png --dataset matterport3d --load_weights_dir /Users/UniFuse-Unidirectional-Fusion/PretrainedModels/

File "/UniFuse-Unidirectional-Fusion/UniFuse/datasets/matterport3d.py", line 83, in __getitem__
    rgb = cv2.cvtColor(rgb, cv2.COLOR_BGR2RGB)
cv2.error: OpenCV(4.5.3) /private/var/folders/24/8k48jl6d249_n_qfxwsl6xvm0000gn/T/pip-req-build-z9mn802i/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

fpath = os.path.join(os.path.dirname(__file__), "datasets", "{}_{}.txt")

But you add txt file with RGB and DEPTH, But shouldn't we just add one RGB image ?

[HalleyJiang]

The evaluate.py is used for performing evaluation on the 4 datasets, not for a single image. If you want to test on an image, you have to modify the code.

[dennisritter]

The --data_path argument should be a directory containing different building ids (e.g. 2t7WUuJeko7, ARNzJeq3xxb, ...), which then each contains a 'pano_skybox_color' directory with images.