- 博客(6)
- 收藏
- 关注
原创 Tensorflow-gpu、python、cuda10.0、cudnn版本对应关系及安装配置
对于版本号大于1.13的tensorflow-gpu版本,如1.14、1.15和2.0,要安装CUDA10.0,不要安装最新的CUDA10.1,安装后会提示缺少很多库文件,而导致GPU版本的tensorflow无法使用
2021-04-30 11:46:24 3284
原创 使用pytorch_yolo_nano进行人体目标检测
代码地址:https://github.com/ardeal/yolo_nano一、准备训练测试集1 将coco数据集转成voc数据集格式,并提取需要的类别(修改路径即可)from pycocotools.coco import COCOimport osimport shutilfrom tqdm import tqdmimport skimage.io as ioimport matplotlib.pyplot as pltimport cv2from PIL import Imag
2021-03-25 16:02:15 532 2
原创 windows10:HRNet+环境配置(pytorch)+网络改动
安装pytorch(CUDA版本:10.0):conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch安装HRNet环境:EasyDict==1.7opencv-python==3.4.1.15shapely==1.6.4Cythonscipypandaspyyamljson_tricksscikit-imageyacs>=0.1.5tensorboardX>=1.6pip install
2020-11-30 22:16:54 4728 23
原创 mmskeleton windows安装(mmdetection)及ST-GCN测试和训练
首先安装mmdetection2:教程环境:OS: Win10VS: 2015mmdetection: 2.1.0mmcv:0.6.0torch=1.3.1torchvision=0.4.2python:3.7.9cuda:10.1一、Pytorch环境安装创建虚拟环境:conda create -n mmd python=3.7conda activate mmdconda install pytorch=1.3 torchvision cudatoolkit=10.1 -c
2020-11-30 22:16:06 5408 47
原创 记录从coco数据集提取语义分割mask的问题
首先下载cocoapi提取mask代码如下(记得修改路径):from pycocotools.coco import COCOimport numpy as npimport skimage.io as ioimport matplotlib.pyplot as pltimport pylabimport cv2import osfrom matplotlib.patches i...
2019-12-02 19:31:19 2265
原创 (真)FCN(Tensorflow):训练自己的数据集
一.下载源码使用原版代码:https://github.com/shekkizh/FCN.tensorflow二.下载网络和数据1.下载VGG网络的权重参数,下载好后的文件路径为./Model_zoo/imagenet-vgg-verydeep-19.mat.网址:http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-ver...
2019-11-18 20:52:53 2707 6
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人