前言
最近在复习yolov5目标检测代码时用了yolov5的最新7.0版本,之前用的是5.0版本,这一新版本相对于之前做了一些提升,对于package的兼容也要好了很多,但也不是说下载了直接就能运行,实际使用过程中还是遇到了许多新的问题,下面就我自己碰到的问题提出解决方法。
问题
wandb是非常好用的可视化工具,但是国内的话,使用时常常会无法同步数据,需要借助魔法来连接服务器,而yolov5的源码恰恰是用了wandb来可视化训练结果,所以在训练时,如果我们的网络不好,就会出现如下所示的问题:
可以看出一直在重连网络,此时我的训练结果已经出来了,但是程序仍在运行中,这对我们后续使用pycharm或者其他工具都很不好。
解决
这里提供一种最简单的方法,那就是暂时取消同步wandb,等网络配置好之后再同步上去。
在train.py的前面几行(加在import导入模块后面就行),添加如下的代码:
os.environ["WANDB_API_KEY"] = YOUR_KEY_HERE
os.environ["WANDB_MODE"] = "offline"
其中,YOUR_KEY_HERE这里要改成你的wandb的key,在这个网址可以找到https://wandb.ai/settings
前提是你要注册一个账号,很简单的,具体在这里找key,如下所示:
复制完key之后,添加代码到train.py即可,具体如下所示:然后再运行train.py就不会一直重连服务器了,并且与运行窗口会给出相应的措施,如果想看可视化结果的话,就等自己的网络好了,或者换一台可以使用的设备,按照它给出的措施进行同步就可以了。
结尾
菜鸟一只,希望这篇文章可以帮助大家,有什么不足请各位大佬批评指正,喜欢的话点个关注哦,后续还会更新yolov5相关内容。