search

zaiweizhang / H3DNet

MIT License
211 stars 25 forks source link

About the PRED_PATH in show_results_sunrgbd.py #8

Closed MezereonXP closed 3 years ago

MezereonXP commented 3 years ago

Hi zaiweizhang,

When I am trying to make visualization on the result of a checkpoint, I found that the PRED_PATH in show_results_sunrgbd.py is confused. After I execute the eval.py, there is no record of predictions.

Could you please give me some advices to set the PRED_PATH or construct the prediction files?

Thanks, Ke

GitBoSun commented 3 years ago

Hi Ke, Thanks for your interest in our paper! When you run the eval.py, you will specify the --dump_dir, which is the folder containing predicted results. The files in that folder will be like 'center'+scan_name+'_nms.npy'. Set the PRED_PATH to this folder and you will load and visualize the predictions. Sorry for the confusion. If you have further questions, please let me know! Best, Bo

MezereonXP commented 3 years ago

Thanks for your reply.

My shell likes:

python eval.py --data_path /data/gaohan/SUN-RGB-D/sunrgbd --dataset sunrgbd --model hdnet --checkpoint_path log_sunrgbd/checkpoint.tar --dump_dir eval_sunrgbd --cluster_sampling seed_fps --use_3d_nms --use_cls_nms --per_class_proposal

But after I execute the shell and specify the --dump_dir, there is only a txt file named log_eval.txt in the dump dir.

MezereonXP commented 3 years ago

I just found the function log_string which can write the string into the log file in eval.py. But I could not find any other functions which save the prediction to the dump dir.

GitBoSun commented 3 years ago

Hi, sorry for the confusion. We forgot to add dump_results to eval.py. We've updated it. Btw, you can use the dump_results function normally in train.py.