Banconxuan / RTM3D

The official PyTorch Implementation of RTM3D and KM3D for Monocular 3D Object Detection
MIT License
453 stars 85 forks source link

无法复现结果 #36

Open cnexah opened 3 years ago

cnexah commented 3 years ago

您好!我使用预训练的model_res18_2.pth,利用src/demo.py进行预测。python3.7, pytorch1.8.1, cuda10.2 命令: python src/demo.py --demo kitti_format/data/kitti/val.txt --data_dir kitti_format --calib_dir kitti_format/data/kitti/calib/ --load_model model_res18_2.pth --gpus 0 --arch res_18 输出: NMS not imported! If you need it, do cd $CenterNet_ROOT/src/lib/external make Fix size testing. training chunk_sizes: [32] The output will be saved to kitti_format/exp/default heads {'hm': 3, 'wh': 2, 'hps': 18, 'rot': 8, 'dim': 3, 'prob': 1, 'reg': 2, 'hm_hp': 9, 'hp_offset': 2} Creating model... => loading pretrained model https://download.pytorch.org/models/resnet18-5c106cde.pth model_res18_2.pth loaded model_res18_2.pth, epoch 199 results_dir kitti_format/exp/results 000554 [2.552675724029541, 1.5896803140640259, -0.03506125509738922] 004822 [2.438274621963501, 1.3753342628479004, -0.11458498984575272] 005190 [2.5769827365875244, 1.5554676055908203, -0.037575557827949524] 005695 [2.445730447769165, 1.4386816024780273, -0.007032535970211029]

测评得到car的结果是这样的: Car AP@0.70, 0.70, 0.70: bbox AP:87.68, 78.48, 69.47 bev AP:21.21, 16.34, 15.16 3d AP:16.47, 13.53, 12.04 aos AP:86.76, 77.13, 68.13 Car AP_R40@0.70, 0.70, 0.70: bbox AP:90.74, 82.73, 71.02 bev AP:15.26, 10.09, 8.57 3d AP:9.70, 6.33, 5.24 aos AP:89.70, 81.12, 69.46

请问我输入的命令是有问题的吗?非常感谢!

GUUUUG commented 3 years ago

我也无法预测得到他给出的效果

DuZzzs commented 3 years ago

同样无法复现结果。

GUUUUG commented 3 years ago

me too

wangziniu1109 commented 3 years ago
  • 下 请问一下你eval的时候报的什么错啊,还有能告诉我你是怎么修改的代码吗
GUUUUG commented 3 years ago

我没有报错啊

------------------ 原始邮件 ------------------ 发件人: @.>; 发送时间: 2021年7月1日(星期四) 晚上9:40 收件人: @.>; 抄送: "王广杰"<953704875@QQ.COM>; @.***>; 主题: Re: [Banconxuan/RTM3D] 无法复现结果 (#36)

下 请问一下你eval的时候报的什么错啊,还有能告诉我你是怎么修改的代码吗

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

tfwang08 commented 3 years ago

请问现在复现出来了吗?我还不行。

GUUUUG commented 3 years ago

我也是,差他的指标一两个点

------------------ 原始邮件 ------------------ 发件人: @.>; 发送时间: 2021年7月4日(星期天) 晚上10:35 收件人: @.>; 抄送: "王广杰"<953704875@QQ.COM>; @.***>; 主题: Re: [Banconxuan/RTM3D] 无法复现结果 (#36)

请问现在复现出来了吗?我还不行。

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

wufeim commented 3 years ago

大无语...