草庐IT

YOLOv8训练自己的数据集(足球检测)

YOLOv8训练自己的数据集(足球检测)前言前提条件实验环境安装环境项目地址LinuxWindows制作自己的数据集训练自己的数据集创建自己数据集的yaml文件football.yaml文件内容进行训练进行验证进行预测数据集获取参考文献前言本文是个人使用YOLOv8训练自己的YOLO格式数据集的应用案例,由于水平有限,难免出现错漏,敬请批评改正。虽然YOLOv8与YOLOv5都是同一个团队Ultralytics发布的,但是YOLOv8的代码封装性比YOLOv5更好。YOLOv8要求的数据集格式与YOLOv5、YOLOv7一致。YOLOv8最大的改变就是抛弃了以往的anchor-base,使用了

狗都能看懂的Pytorch MAML代码详解

目录maml概念数据读取get_file_listget_one_task_data模型训练模型定义源码(觉得有用请点star,这对我很重要~)maml概念首先,我们需要说明的是maml不同于常见的训练方式。以猫狗分类和resnet作为例子,我们将猫狗分类定义为一个task,正常训练一个猫狗分类器,只需要输入猫和狗的图片去训练就好了。所以我们的一个batch中就会有多张猫或者狗的图片,这样训练出来的模型虽说可以预测这张图片是猫还是狗,但要想这个分类器有泛化性,就需要大量猫或狗的图像,而标注大量的数据是要成本的。现在我们假设一个场景,我们没有这么多猫狗分类的数据,但我们有其他task的数据。我们

狗都能看懂的Pytorch MAML代码详解

目录maml概念数据读取get_file_listget_one_task_data模型训练模型定义源码(觉得有用请点star,这对我很重要~)maml概念首先,我们需要说明的是maml不同于常见的训练方式。以猫狗分类和resnet作为例子,我们将猫狗分类定义为一个task,正常训练一个猫狗分类器,只需要输入猫和狗的图片去训练就好了。所以我们的一个batch中就会有多张猫或者狗的图片,这样训练出来的模型虽说可以预测这张图片是猫还是狗,但要想这个分类器有泛化性,就需要大量猫或狗的图像,而标注大量的数据是要成本的。现在我们假设一个场景,我们没有这么多猫狗分类的数据,但我们有其他task的数据。我们

Stable diffusion安装踩坑(win&Mac&iOS)

今天Apple官方支持了这个库,所以特意下载下来试试看,效果还不错,M芯片在深度学习领域第一次给人惊喜了。https://machinelearning.apple.com/research/stable-diffusion-coreml-apple-silicon1.win使用安装分享一下一个有意思的库的安装经历。最近才发现有大神将这种生成模型给开源了,之前一直关注DELL这种的大型模型,被ai这种天马行空的想象力给惊艳到。但他们大都只是api接口,现在贫民百姓也可以自己拥有。下载库去github网址下载->https://github.com/CompVis/stable-diffusio

Stable diffusion安装踩坑(win&Mac&iOS)

今天Apple官方支持了这个库,所以特意下载下来试试看,效果还不错,M芯片在深度学习领域第一次给人惊喜了。https://machinelearning.apple.com/research/stable-diffusion-coreml-apple-silicon1.win使用安装分享一下一个有意思的库的安装经历。最近才发现有大神将这种生成模型给开源了,之前一直关注DELL这种的大型模型,被ai这种天马行空的想象力给惊艳到。但他们大都只是api接口,现在贫民百姓也可以自己拥有。下载库去github网址下载->https://github.com/CompVis/stable-diffusio

深度学习环境配置超详细教程【Anaconda+PyTorch(GPU版)+CUDA+cuDNN】

深度学习环境配置入门深度学习,首先要做的事情就是要搭建深度学习的环境。不管你是Windows用户,Mac用户还是Ubuntu用户,只要电脑配置允许,都可以做深度学习,毕竟Windows、Mac和Ubuntu系统都可以进行深度学习环境的搭建。接下来就记录下自己在Windows系统上搭建深度学习环境的过程,方便自己存档也为大家提供一个参考。本次环境配置主要模式是基于Anaconda+PyTorch(GPU版)+CUDA+cuDNN进行搭建的。所需工具:Python集成开发环境:AnacondaCUDA、cuDNN:英伟达提供的针对英伟达显卡的运算平台。用来提升神经网络的运行效率,如果电脑显卡不满足

深度学习环境配置超详细教程【Anaconda+PyTorch(GPU版)+CUDA+cuDNN】

深度学习环境配置入门深度学习,首先要做的事情就是要搭建深度学习的环境。不管你是Windows用户,Mac用户还是Ubuntu用户,只要电脑配置允许,都可以做深度学习,毕竟Windows、Mac和Ubuntu系统都可以进行深度学习环境的搭建。接下来就记录下自己在Windows系统上搭建深度学习环境的过程,方便自己存档也为大家提供一个参考。本次环境配置主要模式是基于Anaconda+PyTorch(GPU版)+CUDA+cuDNN进行搭建的。所需工具:Python集成开发环境:AnacondaCUDA、cuDNN:英伟达提供的针对英伟达显卡的运算平台。用来提升神经网络的运行效率,如果电脑显卡不满足

基于pytorch搭建多特征LSTM时间序列预测代码详细解读(附完整代码)

文章目录LSTM时间序列预测数据获取与预处理模型构建训练与测试LSTM时间序列预测对于LSTM神经网络的概念想必大家也是熟练掌握了,所以本文章不涉及对LSTM概念的解读,仅解释如何使用pytorch使用LSTM进行时间序列预测,复原使用代码实现的全流程。数据获取与预处理首先预览一下本次实验使用的数据集,该数据集共有三个特征,将最后一列的压气机出口温度作为标签预测(该数据集是我在git上收集到的)定义一个xls文件读取的函数,其中data.iloc()函数是将dataframe中的数据进行切片,返回数据和标签#文件读取defget_Data(data_path):data=pd.read_exc

基于pytorch搭建多特征LSTM时间序列预测代码详细解读(附完整代码)

文章目录LSTM时间序列预测数据获取与预处理模型构建训练与测试LSTM时间序列预测对于LSTM神经网络的概念想必大家也是熟练掌握了,所以本文章不涉及对LSTM概念的解读,仅解释如何使用pytorch使用LSTM进行时间序列预测,复原使用代码实现的全流程。数据获取与预处理首先预览一下本次实验使用的数据集,该数据集共有三个特征,将最后一列的压气机出口温度作为标签预测(该数据集是我在git上收集到的)定义一个xls文件读取的函数,其中data.iloc()函数是将dataframe中的数据进行切片,返回数据和标签#文件读取defget_Data(data_path):data=pd.read_exc

pytorch模型部署小程序

纯纯小白,3月导师叫我去搞个小程序的部署,算法才入门,前端后端毛也不懂的我勇敢冲冲!其实我对模型部署的理解非常模糊,并不清楚它具体在干什么。然后通过一些网络学习,大概理一下思路:先把模型训练好以后,通过优化转化模型、使用推理引擎两步对模型进行infer。推理引擎:已知pytorch框架自带引擎pytorch-mobile,还有专用于移动端推理部署的引擎ncnn(不支持pytorch模型所以需要中间过渡转化)。不过,在后续实践中上述引擎我啥也没用...反正最后呢,我最终花一周时间实现了将pytorch模型在主机跑通,然后通过python的Web应用框架Flask提供后端服务——在局域网下,接收小