- 博客(6)
- 收藏
- 关注
RSS订阅转载 两个Bounding Box的IOU计算代码
Bounding Box的数据结构为（xmin,ymin,xmax,ymax）输入：box1,box2输出：IOU值import numpy as npdef iou(box1,box2): assert box1.size()==4 and box2.size()==4,"bounding box coordinate size must be 4" ...
2019-02-28 23:52:00
1392
转载 非极大值抑制（NMS，Non-Maximum Suppression）的原理与代码详解
1、NMS的原理NMS（Non-Maximum Suppression）算法本质是搜索局部极大值，抑制非极大值元素。NMS就是需要根据score矩阵和region的坐标信息，从中找到置信度比较高的bounding box。NMS是大部分深度学习目标检测网络所需要的，大致算法流程为：1.对所有预测框的置信度降序排序2.选出置信度最高的预测框，确认其为正确预测，并计算他与其他预测...
2019-02-28 23:35:00
518
转载 Pytorch多进程最佳实践
预备知识模型并行（ model parallelism ）：即把模型拆分放到不同的设备进行训练，分布式系统中的不同机器（GPU/CPU等）负责网络模型的不同部分 —— 例如，神经网络模型的不同网络层被分配到不同的机器，或者同一层内部的不同参数被分配到不同机器，如AlexNet的训练。数据并行（ data parallelism ）：即把数据切分，输入到不同的机器有同一个模型的多...
2019-02-20 16:24:00
630
转载 torchvision简介
安装pytorch时，torchvision独立于torch。torchvision包由流行的数据集（torchvision.datasets）、模型架构(torchvision.models)和用于计算机视觉的常见图像转换组成t(torchvision.transforms)。1）数据库MNISTFashion-MNISTKMNISTEMNISTCOCO...
2019-02-20 14:57:00
342
转载 Pytorch数据变换（Transform）
实例化数据库的时候，有一个可选的参数可以对数据进行转换，满足大多神经网络的要求输入固定尺寸的图片，因此要对原图进行Rescale或者Crop操作，然后返回的数据需要转换成Tensor如：import FaceLandmarksDatasetface_dataset = FaceLandmarksDataset(csv_file='data/faces/face_landmar...
2019-02-20 14:35:00
954
转载 Pytorch自定义数据库
1）前言虽然torchvision.datasets中已经封装了好多通用的数据集，但是我们在使用Pytorch做深度学习任务的时候，会面临着自定义数据库来满足自己的任务需要。如我们要训练一个人脸关键点检测算法，提供的训练数据标注如下形式，存在CSV文件中：image_name,part_0_x,part_0_y,part_1_x,part_1_y,part_2_x, ... ...
2019-02-20 11:21:00
239
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人