
pytorch
文章平均质量分 66
Dandelion_2
这个作者很懒,什么都没留下…
展开
-
yolov3 build_targets函数
def build_targets(p, targets, model): # p的shape(YoloLayer_num, batch_size, anchor_num, grid_y, grid_x, xywh+obj_confidence+cls_num) # targets的shape为(num_groundtruth, 6), 其中数字6代表img_index+cls_index+xywh # model为整个yolo的model,以获取当前model对应YoloLayer原创 2021-11-27 14:04:23 · 2557 阅读 · 1 评论 -
江大白共学计划_Pytorch推理及范式_Lesson5
1.课程学习本节课主要对于大白AI课程:https://mp.weixin.qq.com/s/STbdSoI7xLeHrNyLlw9GOg《Pytorch推理及范式》课程中的第五节课进行学习。2.作业题目必做题:(1) 自己找 2 张其他图,用 Yolox_s 进行目标检测,并注明输入尺寸和两个阈值。两张图片的输入尺寸都是640*640,两个阈值分别是: conf_threshold = 0.6 nms_threshold = 0.45思考题:(1) Yolox_s:用 time 模块和原创 2021-10-25 18:31:59 · 294 阅读 · 0 评论 -
江大白共学计划_Pytorch推理及范式_Lesson4
1.课程学习本节课主要对于大白AI课程:https://mp.weixin.qq.com/s/STbdSoI7xLeHrNyLlw9GOg《Pytorch推理及范式》课程中的第四节课进行学习。2.作业题目必做题:(1) 对 “./images/car.jpg” 做语义分割,提取出里面的车辆,模仿上课时,对“可视化推理结果”和“BGRA 四通道图”进行保存。可视化推理结果:“BGRA"四通道图像:(2) 自己找 2 张其他图,对图中某个类别进行分割,并保存“BGRA 四通道图”。图1:i原创 2021-10-22 17:20:10 · 268 阅读 · 0 评论 -
江大白共学计划_Pytorch推理及范式_Lesson3
文章目录1.课程学习2.作业题目必做题:思考题:3 课程总结:1.课程学习本节课主要对于大白AI课程:https://mp.weixin.qq.com/s/STbdSoI7xLeHrNyLlw9GOg《Pytorch推理及范式》课程中的第二节课进行学习。2.作业题目必做题:(1) 把模型改为resnet18,加载相应的模型权重(Lesson2的物料包中有),跑一下0.jpg和1.jpg,看一下输出结果。结果:(2) 自己找2张其他图,用resnet18做下推理。图片:结果:原创 2021-10-19 14:19:38 · 317 阅读 · 1 评论 -
江大白共学计划_Pytorch推理及范式_Lesson2
江大白共学计划_Pytorch推理及范式_Lesson2 课程作业必做题:(1) 从torchvision中加载resnet18模型结构,并载入预训练好的模型权重 ‘resnet18-5c106cde.pth’ (在物料包的weights文件夹中)。import torchimport torchvision.models as model# 从torchvision加载res18模型结构model = model.resnet18()# 加载权重pretrained_state原创 2021-10-15 19:53:23 · 316 阅读 · 0 评论 -
torch.randn
torch.randn(*sizes, out=None) → Tensor(张量)返回一个张量,包含了从标准正态分布(均值为0,方差为 1)中抽取一组随机数,形状由可变参数sizes定义。 参数:sizes (int…) – 整数序列,定义了输出形状out (Tensor, optinal) - 结果张量二维>>> import torch>>> ...原创 2019-10-19 14:23:23 · 1057 阅读 · 0 评论 -
python中的list.sort()与sorted()区别于应用
sort是容器的函数:sort(cmp=None, key=None, reverse=False)sorted是python的内建函数:sorted(iterable, cmp=None, key=None, reverse=False)参数解析:cmp:比较函数,比较什么参数由key决定。例如:cmp(e1, e2) 是具有两个参数的比较函数,返回值:负数(e1 < e2);0(e...原创 2019-10-13 09:20:19 · 190 阅读 · 0 评论 -
Python中os.mkdir()与os.makedirs()的区别及用法
os.mkdir()是一级一级的创建目录,import os os.mkdir('d:\hello') # 正常os.mkdir('d:\hello\hi') # 正常 # 如果d:\hello目录不存在# 则os.mkdir('d:\hello\hi')执行失败如果上一级的目录不存在就会报错。os.makedirs()可以递归创建目录,就是一次可以创建多级目录,...原创 2019-10-12 21:57:36 · 466 阅读 · 0 评论 -
PyTorch常用的torchvision transforms函数
from torchvision import transforms# Define a transform to do the normalization of the datatransform_data = transforms.Compose([transforms.ToTensor(), transforms.Normali...原创 2019-10-09 15:25:12 · 804 阅读 · 0 评论 -
pytorch构造一个小型cnn
torch.nn只接受mini-batch的输入,也就是说我们输入的时候是必须是好几张图片同时输入。例如:nn. Conv2d 允许输入4维的Tensor:n个样本 x n个色彩频道 x 高度 x 宽度#coding=utf-8import torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.autog...转载 2019-10-07 21:18:39 · 458 阅读 · 0 评论