deepsnake 安装及训练步骤

2024-03-25 08:10
文章标签 安装 步骤 训练 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.7conda 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

 

这篇关于deepsnake 安装及训练步骤的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/844366

相关文章

Nginx设置连接超时并进行测试的方法步骤

《Nginx设置连接超时并进行测试的方法步骤》在高并发场景下,如果客户端与服务器的连接长时间未响应,会占用大量的系统资源,影响其他正常请求的处理效率,为了解决这个问题,可以通过设置Nginx的连接... 目录设置连接超时目的操作步骤测试连接超时测试方法:总结:设置连接超时目的设置客户端与服务器之间的连接

Python使用国内镜像加速pip安装的方法讲解

《Python使用国内镜像加速pip安装的方法讲解》在Python开发中,pip是一个非常重要的工具,用于安装和管理Python的第三方库,然而,在国内使用pip安装依赖时,往往会因为网络问题而导致速... 目录一、pip 工具简介1. 什么是 pip?2. 什么是 -i 参数?二、国内镜像源的选择三、如何

Debezium 与 Apache Kafka 的集成方式步骤详解

《Debezium与ApacheKafka的集成方式步骤详解》本文详细介绍了如何将Debezium与ApacheKafka集成,包括集成概述、步骤、注意事项等,通过KafkaConnect,D... 目录一、集成概述二、集成步骤1. 准备 Kafka 环境2. 配置 Kafka Connect3. 安装 D

Spring AI集成DeepSeek的详细步骤

《SpringAI集成DeepSeek的详细步骤》DeepSeek作为一款卓越的国产AI模型,越来越多的公司考虑在自己的应用中集成,对于Java应用来说,我们可以借助SpringAI集成DeepSe... 目录DeepSeek 介绍Spring AI 是什么?1、环境准备2、构建项目2.1、pom依赖2.2

Goland debug失效详细解决步骤(合集)

《Golanddebug失效详细解决步骤(合集)》今天用Goland开发时,打断点,以debug方式运行,发现程序并没有断住,程序跳过了断点,直接运行结束,网上搜寻了大量文章,最后得以解决,特此在这... 目录Bug:Goland debug失效详细解决步骤【合集】情况一:Go或Goland架构不对情况二:

Python安装时常见报错以及解决方案

《Python安装时常见报错以及解决方案》:本文主要介绍在安装Python、配置环境变量、使用pip以及运行Python脚本时常见的错误及其解决方案,文中介绍的非常详细,需要的朋友可以参考下... 目录一、安装 python 时常见报错及解决方案(一)安装包下载失败(二)权限不足二、配置环境变量时常见报错及

SpringBoot+MyBatis-Flex配置ProxySQL的实现步骤

《SpringBoot+MyBatis-Flex配置ProxySQL的实现步骤》本文主要介绍了SpringBoot+MyBatis-Flex配置ProxySQL的实现步骤,文中通过示例代码介绍的非常详... 目录 目标 步骤 1:确保 ProxySQL 和 mysql 主从同步已正确配置ProxySQL 的

nginx部署https网站的实现步骤(亲测)

《nginx部署https网站的实现步骤(亲测)》本文详细介绍了使用Nginx在保持与http服务兼容的情况下部署HTTPS,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值... 目录步骤 1:安装 Nginx步骤 2:获取 SSL 证书步骤 3:手动配置 Nginx步骤 4:测

Springboot 中使用Sentinel的详细步骤

《Springboot中使用Sentinel的详细步骤》文章介绍了如何在SpringBoot中使用Sentinel进行限流和熔断降级,首先添加依赖,配置Sentinel控制台地址,定义受保护的资源,... 目录步骤 1: 添加 Sentinel 依赖步骤 2: 配置 Sentinel步骤 3: 定义受保护的

MySQL8.2.0安装教程分享

《MySQL8.2.0安装教程分享》这篇文章详细介绍了如何在Windows系统上安装MySQL数据库软件,包括下载、安装、配置和设置环境变量的步骤... 目录mysql的安装图文1.python访问网址2javascript.点击3.进入Downloads向下滑动4.选择Community Server5.