Akangman的博客

原创 用YOLOv5训练自己的数据集

1、环境配置，他在B站中用的是cpu版的pytorch，完完全全的按照他的步骤一步一步来，最后环境没有配置成功。首先还是的找到自己电脑的版本驱动，在搜索中输入NVIDAI控制面板，点击系统信息，点击组件就能看到驱动版本了。2、数据集的大小，刚开始我导入的数据集比较大，总是出错，原因是有的图片格式不对，然后选取了一部分图片，这个错误解决了。在下载pytorch之前，可以在终端激活虚拟环境后，导入镜像源，在将对应的代码输入到终端。3、代码运行错误，其中有一个np.int，把代码中对应的这些改为int。