pytorch
牙疼_LG
深度学习,计算机视觉,目标检测,点云分割.
https://github.com/yatengLG
展开
-
Pytorch-SSD目标检测算法(Single Shot MultiBox Detector)(简单,明了,易用,全中文注释,单机多卡训练,训练过程可视化,视频检测)
SSD-Pytorch作者为pytorch中文网维护者之一,网站会发许多深度学习相关文章,欢迎pytorch使用者多多支持.项目Github地址:Github 欢迎star, fork一. 项目简介项目是在实际项目上整理精简而来,主要针对单机多卡环境实现.项目整体结构清晰明了,对训练过程在不失灵活性的前提下进行了包装.训练,测试,检测过程均只需简单几行即可实现.如需深入了...原创 2019-11-22 15:29:42 · 1986 阅读 · 2 评论 -
Retinanet 基于Pytorch实现,针对单机多卡训练环境. (全中文注释,结构清晰,目标检测,视频检测)
Retinanet-Pytorch作者为pytorch中文网维护者之一,网站会发许多深度学习相关文章,欢迎pytorch使用者多多支持.项目Github地址:Github 欢迎star, fork一. 项目简介该项目是在SSD-Pytorch项目基础上修改而来.相关文章见:针对单机多卡环境的SSD目标检测算法实现(Single Shot MultiBox Detector)(简单,明了,...原创 2019-11-22 15:23:29 · 1198 阅读 · 4 评论 -
Focal loss-Pytorch Retinanet损失函数pyotrch实现.
Retinanet作者为pytorch中文网维护者之一,网站会发许多深度学习相关文章,欢迎pytorch使用者多多支持.项目Github地址:Github 欢迎star, forkretinanet是ICCV2017的Best Student Paper Award(最佳学生论文),何凯明是其作者之一.文章中最为精华的部分就是损失函数 Focal loss的提出.论文中提出类别失衡是造成...原创 2019-11-22 15:16:30 · 1463 阅读 · 2 评论 -
pytorch 编译后,在C++环境中运行
一. 准备工作1.1 首先将要转换的pytorch模型转为TorchScript模型这里推荐使用追踪的方法进行转换import torchfrom SSD.Build_ssd import SSDfrom Configs import _C as cfg# 先初始化模型model = SSD(cfg)model.eval()# 导入模型参数model.load_sta...原创 2019-10-28 13:51:18 · 1995 阅读 · 0 评论 -
C++扩展pytorch,简单加法例子
首先编写 our_add.cpp, 引入torch.h 定义加法#include <torch/torch.h>#include <vector>#include <iostream>torch::Tensor ADD(const torch::Tensor& a, const torch::Tensor& b){ ret...原创 2019-10-23 17:24:13 · 995 阅读 · 2 评论 -
Focal loss 基于pytorch实现.
Focal_loss作者是一名深度学习工程师,主要研究计算机视觉与三维点云处理,作者Github:Github.欢迎多多交流.同时作者也是pytorch中文网的维护者之一,网站会不定时更新深度学习相关文章,欢迎大家多多支持.retinanet是ICCV2017的Best Student Paper Award(最佳学生论文),何凯明是其作者之一.文章中最为精华的部分就是损失函数 Foc...原创 2019-09-18 09:16:08 · 3643 阅读 · 1 评论 -
多人同时使用 服务器 anaconda 环境,测试程序
一,服务器端 安装并开启 sshsudo apt-get install openssh-server然后,你可以通过以下命令:来查看 ssh 是否开启ps -s|grep ssh如图,出现ssh 即表明ssh 服务器已开启,如果没有开启,可以使用以下命令开启sudo service ssh start然后你可以通过 以下命令来查看 你的ip:ifco...原创 2019-03-31 19:55:54 · 3114 阅读 · 0 评论 -
ubuntu下 创建多人共用的anaconda环境
一.Anaconda下载地址:https://www.anaconda.com/download/#linux然后用 bash 安装下载好的 ".sh" 安装包;到 安装路径时,指定 到/usr/local/目录下 (linux下安装第三方多用户使用的共享软件一般都按在 /usr/local 目录下) 二.配置 环境变量打开 /etc/profile 文件,在该文件最后加入这...原创 2018-11-04 15:58:09 · 6293 阅读 · 0 评论 -
pytorch下大型数据集(大型图片)的导入
使用torch.utils.data.Dataset类 处理图片数据时,1. 我们需要定义三个基本的函数,以下是基本流程class our_datasets(Data.Dataset): def __init__(self,root,is_resize=False,is_transfrom=False): #这里只是个参考。按自己需求写。 sel...原创 2018-09-06 17:03:22 · 6519 阅读 · 4 评论 -
ubuntu18.04+windows10 双系统下,cuda9.0+cudnn7.2.1+pytorch-gpu版本安装流程
博主是在笔记本上装的,在原有win10的情况下装的双系统ubuntu18.04+windows10。笔记本是双显卡,独显为gtx960m。#######################################################################################(原创不易,转载请注明 且附本文链接。)##################...原创 2018-08-31 22:23:28 · 2139 阅读 · 2 评论