| |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
-> 人工智能 -> deepsnake 安装及训练步骤 -> 正文阅读 |
|
[人工智能]deepsnake 安装及训练步骤 |
安装环境: 操作系统:ubantu 16.04 python 版本:3.7 pytorch+cuda版本: pytorch1.1+cuda10.0 https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl 注:用命令nvcc -V 查看显卡cuda版本,pytorch+cuda版本需要与显卡的cuda版本一致 首先创建虚拟环境,分别执行以下命令:
若torch-1.1.0-cp37-cp37m-linux_x86_64.whl 放在/root文件夹下 ,则执行
然后把snake-master.zip解压到根目录/root,接着分别执行
我的cocoval.py文件在/root/miniconda3/envs/snake/lib/python3.7/site-packages/pycocotools 目录下: 接下来修改cocoeval.py,参考文章:https://blog.csdn.net/qq_41821678/article/details/105643630 最后训练,执行下面命令后等待下载好dla34-ba72cf86.pth 就会开始训练了
?训练生成的模型在 data/model/snake/fish文件夹中 ? 利用生成的模型对图片进行处理:
常见bug及解决方案汇总:1. 数据集路径有问题,检查一下.json文件或dataset_catalog.py文件。注意:.json文件中出现的文件名,数据集中一定要有并且路径正确,否则会报错 2. 重新执行一遍命令 或 删除data文件夹里面的 model、record、result文件夹。/root/.local/share/Trash是 回收站目录,如果是图像界面删除文件,要到这个目录用 rm -rf 命令删除才会释放空间 3. pip install --upgrade numpy 4. 可以把batch-size设为1 ? |
|
|
上一篇文章 下一篇文章 查看所有文章 |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 | -2024/11/27 12:45:28- |
|
网站联系: qq:121756557 email:121756557@qq.com IT数码 |