AugFPN：训练出现RuntimeError: Expected object of scalar type Byte but got scalar type Bool for argument

最新推荐文章于 2024-04-12 18:05:34 发布

White_yn

最新推荐文章于 2024-04-12 18:05:34 发布

阅读量668

点赞数 2

本文链接：https://blog.csdn.net/White_yn/article/details/119869291

版权

刚安装完AugFPN，训练时出现错误

File “/data/data/ysc_data/project/Pedestron/mmdet/core/anchor/anchor_target.py”, line 176, in anchor_inside_flags
(flat_anchors[:, 2] < img_w + allowed_border) &
RuntimeError: Expected object of scalar type Byte but got scalar type Bool for argument #2 ‘other’ in call to _th_and

转载：https://github.com/open-mmlab/mmdetection/issues/3587

将对应的mmdet/core/anchor/anchor_target.py中 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-MBk3kcBF-1629704504416)(https://img-blog.csdnimg.c/b90d0bdd42a340e9ba8b7b72c56ee328.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1doaXRlX3lu,size_16,color_FFFFFF,t_70)]
改为
在这里插入图片描述