用pytorch实现ssd并在自己的数据集上进行行人检测
2024-01-24 14:36:14 107KB pytorch ssd
1
生成绘画火炬 根据作者的,对PyTorch重新。 先决条件 该代码已经在Ubuntu 14.04上进行了测试,以下是需要安装的主要组件: Python3 PyTorch 1.0+ 火炬视觉0.2.0+ 张量板 pyyaml 训练模型 python train.py --config configs/config.yaml 检查点和日志将保存到checkpoints 。 用训练好的模型进行测试 默认情况下,它将在检查点中加载最新保存的模型。 您也可以使用--iter通过迭代选择保存的模型。 训练有素的PyTorch模型:[ ] [] python test_single.py \ --image examples/imagenet/imagenet_patches_ILSVRC2012_val_00008210_input.png \ --mask examples/cen
1
舌苔数据集,两千多张图片,512x512通道,包含原图和labelme打好的标签
2024-01-22 16:30:28 206.13MB 数据集 python pytorch tensorflow
1
基于Pytorch实现GRU模型
2024-01-18 16:17:36 321KB pytorch 深度学习
1
本文介绍了使用pytorch2.0进行图像分类的实战案例,包括数据集的准备,卷积神经网络的搭建,训练和测试的过程,以及模型的保存和加载。本案例使用了CIFAR-10数据集,包含10个类别的彩色图像,每个类别有6000张图像,其中5000张用于训练,1000张用于测试。本案例使用了一个简单的卷积神经网络,包含两个卷积层和两个全连接层,使用ReLU激活函数和交叉熵损失函数,使用随机梯度下降优化器。本案例可以在GPU和CPU上运行,根据设备的不同自动切换。本案例适合入门pytorch深度学习和练手,也可以用到项目当中。代码精炼,容易修改进行二次完善和开发。
2024-01-16 14:08:43 325.06MB pytorch 数据集 计算机视觉
1
,机器学习应用程序的广泛部署激发了人们对利用存储在移动设备上的大量数据的兴趣。为了保护数据隐私,联邦学习被提出通过在参与设备上执行本地分布式训练并将本地模型聚合为全局模型来学习共享模型。然而,由于移动设备的网络连接有限,联邦学习在所有参与设备上并行执行模型更新和聚合是不切实际的。此外,跨所有设备的数据样本通常不是独立同分布的(IID),这对联邦学习的收敛性和速度提出了额外的挑战。 在本文中,我们提出了一个经验驱动的控制框架FAVOR,它可以智能地选择客户端设备参与每一轮联邦学习,以抵消非iid数据引入的偏差,并加快收敛速度。通过实证和数学分析,我们观察到设备上训练数据的分布与基于这些数据训练的模型权值之间存在隐式联系,这使我们能够根据该设备上上传的模型权值来描述该设备上的数据分布。然后,我们提出了一种基于深度q学习的机制,该机制学习在每个通信轮中选择一个设备子集,以最大限度地奖励,鼓励提高验证准确性,并惩罚使用更多通信轮。通过在PyTorch中进行的大量实验,我们表明,与联邦平均算法相比,联邦学习所需的通信轮数在MNIST数据集上最多可以减少49%。
2024-01-15 17:58:33 1.13MB pytorch pytorch
1
本资源为Vitis-AI3.0版本docker镜像的.tar文件的下载链接,适用于NVIDIA显卡硬件平台,内置pytorch量化编译器镜像以及pytorch优化器镜像 使用方法: 使用docker load指令将镜像文件导入后(导入后可以使用docker tag指令改名),再按照官方手册中的使用方法即可 docker镜像生成过程: 按照官方github提供的3.0.0.001版本源代码中的dockerfile进行docker创建 以NVIDIA提供的nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04镜像为基础 仅修改apt、python、conda为国内下载源,其他未作变动 dockerfile修改内容: 参考文章https://blog.csdn.net/qq_36745999/article/details/129920225
2024-01-15 17:47:55 78B pytorch docker Vitis-AI Vitis
1
演讲 语音是一个开放源代码包,用于构建用于自动语音识别的端到端模型。 当前支持关注的序列到序列模型,连接器时间分类和RNN序列转换器。 该软件的目的是促进语音识别的端到端模型的研究。 这些模型在PyTorch中实现。 该软件仅在Python3.6中经过测试。 我们不会为Python2.7提供向后兼容性。 安装 我们建议创建一个虚拟环境并在其中安装python要求。 virtualenv source /bin/activate pip install -r requirements.txt 然后按照适用于您的计算机的版本的安装说明进行操作。 安装所有python需求后,从顶层目录运行: make 构建过程需要CMake以及Make。 之后,从仓库根目录获取setup.sh 。 source setup
2024-01-15 10:50:42 112KB Python
1
pytorch深度学习图片风格迁移项目源码+资料,代码注解非常详细,适合新手学习。
2024-01-13 16:13:43 8.35MB pytorch pytorch 深度学习
1
NLP-study 记录做过的NLP任务,包含但不限于文本分类,关系分类,命名实体识别,文本摘要,文本生成等,基于tensorflow2.0或者pytorch框架。
2024-01-12 21:57:28 83.48MB Python
1