在服务器训练CenterFusion网络时报错

Question

Traceback (most recent call last):
  File "main.py", line 140, in <module>
    main(opt)
  File "main.py", line 97, in main
    log_dict_val, preds = trainer.val(epoch, val_loader)
  File "/root/autodl-tmp/CenterFusion/src/lib/trainer.py", line 403, in val
    return self.run_epoch('val', epoch, data_loader)
  File "/root/autodl-tmp/CenterFusion/src/lib/trainer.py", line 178, in run_epoch
    output, loss, loss_stats = model_with_loss(batch, phase)
  File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
    return forward_call(*input, **kwargs)
  File "/root/autodl-tmp/CenterFusion/src/lib/trainer.py", line 123, in forward
    outputs = self.model(batch['image'], pc_hm=pc_hm, pc_dep=pc_dep, calib=calib)
  File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
    return forward_call(*input, **kwargs)
  File "/root/autodl-tmp/CenterFusion/src/lib/model/networks/base_model.py", line 110, in forward
    pc_hm = generate_pc_hm(z, pc_dep, calib, self.opt)
  File "/root/autodl-tmp/CenterFusion/src/lib/utils/pointcloud.py", line 273, in generate_pc_hm
    pc_dep_to_hm_torch(pc_hm[i], pc_dep_b, depth, bbox, dist_thresh, opt)
  File "/root/autodl-tmp/CenterFusion/src/lib/utils/pointcloud.py", line 282, in pc_dep_to_hm_torch
    bbox_int = torch.tensor([torch.floor(bbox[0]), 
TypeError: only integer tensors of a single element can be converted to an index

Answer 1

• 请看👉 ：深入浅出CenterFusion