- 博客(13)
- 收藏
- 关注
RSS订阅原创 PyTorch使用指定GPU进行训练
在命令行窗口输入以下指令：CUDA_VISIBLE_DEVICES=2 python train.pyCUDA_VISIBLE_DEVICES用来指定GPU的编号，从0开始计数。
2019-09-21 10:43:51
1177
原创 One-Stage目标检测算法之——CornerNet
论文地址：https://arxiv.org/abs/1808.01244代码地址：https://github.com/princeton-vl/CornerNet创新点：提出了一种通过预测一对关键点（左上角和右下角）来预测目标 bounding box 的方法。 提出了一种新的池化方法 corner pooling，该池化方法用来更好的定位 bounding box...
2019-07-09 17:06:46
829
原创 Detectron：用自己的数据集（VOC格式）训练RetinaNet
一、数据集（VOC）Detectron天然支持COCO格式的数据集，但是可能我们手中的数据集是VOC格式的，这就需要我们手动的对VOC数据集进行相应的转换（xml→json），下面就以我自己的一个数据集为例，逐步说明一下数据集的配置过程。1. 检查数据集（如果确定自己的数据集没有问题，可以跳过这一步）首先，我们应当先检查自己的数据集是否完整，这一步也很重要，因为我就是在这一步中浪费了...
2019-06-20 16:37:25
1867
3
原创 用darknet框架中的YOLOv3对自己的数据集进行训练和预测
Referencehttps://pjreddie.com/darknet/yolo/https://blog.csdn.net/weixin_42731241/article/details/81352013https://blog.csdn.net/mieleizhi0522/article/details/79989754https://blog.csdn.net/yinh...
2019-05-07 17:03:48
2448
3
原创 安装jupyter出现ipython与prompt-toolkit要求版本不一致的问题
使用 sudo pip install jupyter 时，遇到了以下报错：解决方法：使用sudo pip install jupyter-console==5.2.0降低jupyter-console的版本。测试有效Referencehttps://www.jianshu.com/p/bd3a1824854d...
2019-04-25 22:38:30
5103
原创 docker attach 卡住的问题
在将image镜像run成容器时使用以下命令sudo NV_GPU=1 nvidia-docker run --name *** -d -p ****:5000 -it -v **/**:/workspace tensorflow:18.09-py3 之前是一直在最后加上 bash 的，但这一次在使用digits镜像时，提示bash是一个无法识别的参数，所以在run成容器时，将命令中的b...
2019-04-20 11:55:09
7888
1
转载 caffe2_tutorials中的图像预处理代码
需要学习的是：1.将HWC转变为CHW 。如果我们使用的是GPU，就需要将图片的维度转变为CHW，因为这是cuDNN所期望的维度顺序，这样可以加速网络训练的过程。# switch to CHW (HWC --> CHW)img = img.swapaxes(1, 2).swapaxes(0, 1)2.将RGB转变为BGR。因为Caffe中支持的图片通道顺序是BG...
2019-04-16 21:12:37
239
原创 caffe2官方文档Loading_Pretrained_Models中，读取网络参数pb文件时的RuntimeError解决方法
在Loading_Pretrained_Models.ipynb中，导入预训练参数和网络结构的程序段如下：# Read the contents of the input protobufs into local variableswith open(INIT_NET, "rb") as f: init_net = f.read()with open(PREDICT_NET, "...
2019-04-15 09:43:05
1408
原创 python求取一个数组的（最大值/最小值）及其（索引）
x = [3, 2.2, 7.4, 6, 4]min_index, min_number = min(enumerate(x), key=operator.itemgetter(1))# 输出 1 2.2max_index, max_number = max(enumerate(x), key=operator.itemgetter(1))# 输出 2 7.4解析：en...
2019-04-10 22:01:56
71837
3
转载 C++数学运算 & 随机数生成
数学运算C++内置了丰富的数学函数，可以对各种数字进行运算，下表列出了C++中一些有用的数学函数。为了利用这些函数，需要引用数学头文件<cmath>#include <cmath>序号 函数&描述 1 double cos(double); 该函数返回弧度角（double 型）的余弦。 2 doub...
2019-04-10 18:14:32
788
原创 修改Darknet源码，使其可以保存Bounding-box坐标信息
因为使用Darknet进行目标检测任务时，框架只会输出图片形式的检测结果（即将bounding-box显式的画在原图上），但并不会输出单独的坐标信息，这给我们的进一步分析带来很多不便。所以我们就要对框架源码进行一定的修改，使其能保存图片中目标的bounding-box坐标到txt文件中。Step 1首先我们要找到源码中画框的函数，因为画框是需要坐标信息的，我们可以在这个函数中找到bound...
2019-04-07 21:55:18
4295
21
翻译 YOLOv2：Better，Faster(2016,12)
YOLO9000：Better，Faster，Stronger摘要：YOLO9000是一个可以检测超过9000类目标的实时目标检测系统提出了一种联合训练的方法：可以同时在COCO检测数据集和ImageNet分类数据集上训练，所以可以正确预测在detection数据集上没有labelled的目标类。总之，这篇文章的重点是：充分利用已有的大型分类数据集来扩充目标检测的数据集，目标检测数据集用来...
2019-03-29 16:47:43
158
转载 TFRecords文件的生成和读取
转载自 https://blog.csdn.net/qq_36330643/article/details/77366083关于TensorFlow读取数据，官网提供了3种方法： 1）Feeding ： 在TensorFlow程序运行的每一个epoch，用python代码在线提供数据 2）Reader ： 在一个计算图（tf.graph）的开始前，将文件读入到流（queue）中 3）在...
2018-08-18 20:14:51
406
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人