- 博客(2)
- 收藏
- 关注
RSS订阅原创 AttributeError: ‘Tensor‘ object has no attribute ‘bool‘
这个问题是跑b站作者bubbliiiing写的YOLOv4（pytorch版本）代码中出现的问题，具体是计算CIoU的过程中需要将mask的值转化为bool型，作者直接采用的mask.bool，ciou = ( 1 - box_ciou( pred_boxes_for_ciou[mask.bool], t_box[mask.bool] ) ) * box_loss_scale[mask.bool]这个也是版本的问题，将pytorch版本换到1.2.0以上应该就可以解决，不过我的9.0版本的显卡最高也就只
2020-11-03 16:21:18
2321
5
原创 TypeError: cat() got an unexpected keyword argument ‘axis‘
这个问题是跑b站作者bubbliiiing写的YOLOv4（pytorch版本）代码中出现的问题，在yolo4.py中有一个特征层之间的连接过程P4 = torch.cat( [P4,P5_upsample], axis=1)，作者给出的常见问题汇总给出的建议是将pytorch版本更新到1.2.0以上，但是我的显卡是老显卡，cuda版本9.0，查看了版本对应表，最高也只能安装1.0.1版本，但是对于旧版本应该也是可以用的，只要将axis去掉或者改为dim就可以了，具体为P4 = torch.cat(
2020-11-03 15:55:09
1959
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人