ethnhe / FFB6D

[CVPR2021 Oral] FFB6D: A Full Flow Bidirectional Fusion Network for 6D Pose Estimation.
MIT License
296 stars 71 forks source link

Error in test_ycb and test_lm #78

Open HannahHaensen opened 2 years ago

HannahHaensen commented 2 years ago

After downloading and installing I just run the demo and test files.

I encounter the following error:

Traceback (most recent call last):
  File "train_lm.py", line 686, in <module>
    train()
  File "train_lm.py", line 667, in train
    val_loss, res = trainer.eval_epoch(
  File "train_lm.py", line 404, in eval_epoch
    seg_res_fn += '_%s%.2f' % (k, v)
TypeError: must be real number, not list

https://github.com/ethnhe/FFB6D/issues/50 in this issue there is no fix provided

??? is this a valid fix ??? : seg_res_fn += '_%s%.2f' % (k, np.array(v).mean() * 100)

cause otherwise the filename of the log file will be endlessly ?

nachi9211 commented 2 years ago

This is what I am implementing while dealing with this during both training and testing phases. However is anyone can provide a definite answer or explanation it would help. I do not understand the dimensions of involved list personally.

2022Nison commented 2 years ago

maby you can edit on train.py,you can try to edit 'print(k,v[0])' instead of 'print(k, v)'