解决pytorch-yolov3 train 报错的问题

 更新时间:2020年02月18日 10:06:57   作者:bbzz2  
今天小编就为大家分享一篇解决pytorch-yolov3 train 报错的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead

https://github.com/eriklindernoren/PyTorch-YOLOv3/blob/master/models.py#L191

将model.py 

obj_mask转为int8 bool

obj_mask=obj_mask.bool() # convert int8 to bool

noobj_mask=noobj_mask.bool() #convert int8 to bool 

以上这篇解决pytorch-yolov3 train 报错的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • flask项目集成swagger的方法

    flask项目集成swagger的方法

    这篇文章主要介绍了flask项目集成swagger的方法,帮助大家更好的理解和使用flask框架,感兴趣的朋友可以了解下
    2020-12-12
  • Restful_framework视图组件代码实例解析

    Restful_framework视图组件代码实例解析

    这篇文章主要介绍了Restful_framework视图组件代码实例解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-11-11
  • Python3内置json模块编码解码方法详解

    Python3内置json模块编码解码方法详解

    Python3中我们利用内置模块json解码和编码JSON对象。json模块提供了四个功能:dumps、dump、loads、load本文详细讲解了Python3内置json模块的详细使用方法
    2021-10-10
  • 对python 命令的-u参数详解

    对python 命令的-u参数详解

    今天小编就为大家分享一篇对python 命令的-u参数详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-12-12
  • 详解pandas.DataFrame中删除包涵特定字符串所在的行

    详解pandas.DataFrame中删除包涵特定字符串所在的行

    这篇文章主要介绍了pandas.DataFrame中删除包涵特定字符串所在的行,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-04-04
  • python中sub-pub机制实现Redis的订阅与发布

    python中sub-pub机制实现Redis的订阅与发布

    本文主要介绍了python中sub-pub机制实现Redis的订阅与发布,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2023-03-03
  • Python可以实现栈的结构吗

    Python可以实现栈的结构吗

    在本篇文章里小编给各位整理的是关于Python实现栈的结构的条件的相关知识点,有需要的朋友们可以学习下。
    2020-05-05
  • 解决webdriver.Chrome()报错:Message:''chromedriver'' executable needs to be in Path

    解决webdriver.Chrome()报错:Message:''chromedriver'' executable n

    这篇文章主要介绍了解决webdriver.Chrome()报错:Message:'chromedriver' executable needs to be in Path ,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-06-06
  • python3+PyQt5实现自定义分数滑块部件

    python3+PyQt5实现自定义分数滑块部件

    这篇文章主要为大家详细介绍了python3+PyQt5实现自定义分数滑块部件,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-04-04
  • python numpy中array与pandas的DataFrame转换方式

    python numpy中array与pandas的DataFrame转换方式

    这篇文章主要介绍了python numpy中array与pandas的DataFrame转换方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2022-07-07

最新评论