IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: 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版本一致

首先创建虚拟环境,分别执行以下命令:

conda create -n snake python=3.7

conda activate snake

若torch-1.1.0-cp37-cp37m-linux_x86_64.whl 放在/root文件夹下 ,则执行

pip install /root/torch-1.1.0-cp37-cp37m-linux_x86_64.whl

然后把snake-master.zip解压到根目录/root,接着分别执行

cd /root/snake-master
pip install cython
pip install termcolor
pip install -r requirements.txt

我的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 就会开始训练了

python train_net.py --cfg_file configs/fish.yaml model fish

?训练生成的模型在 data/model/snake/fish文件夹中

?

利用生成的模型对图片进行处理:

python? run.py --type visualize --cfg_file configs/fish.yaml test.dataset fishval ct_score 0.3 model xxx? (可选)

常见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

?

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2021-10-01 16:51:20  更:2021-10-01 16:54:27 
 
开发: 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年5日历 -2024/5/21 20:44:11-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码