摘要: One-stage检测器 几种比较常见的one-stage检测器有YOLO系列和Retinanet等。接下来简要介绍一下这两种结构。 YOLO Retinanet retinanet主要由三部分组成,分别是backbone、FPN和检测器头 1、backbone 在retinanet模型中的back 阅读全文
posted @ 2022-03-25 11:06 maple_hx 阅读(571) 评论(0) 推荐(0)
摘要: torch.Tensor 基础 Tensor就是pytorch中存储数据的主要格式,跟numpy类似 这里,我们先介绍一些最基本的操作和常用的功能 在numpy中,我们是通过shape来获取数组的形状,而在我们的tensor当中,我们使用size来得到形状。 有时候,我们需要对数组形状进行改变,我们 阅读全文
posted @ 2022-03-17 14:28 maple_hx 阅读(61) 评论(0) 推荐(0)
摘要: torch.min()、torch.max()、torch.prod() 这两个函数很好理解,就是求张量中的最小值和最大值以及相乘 1.在这两个函数中如果没有指定维度的话,那么默认是将张量中的所有值进行比较,输出最大值或者最小值或是所有值相乘。 2.而当指定维度之后,会将对应维度的数据进行比较,同时 阅读全文
posted @ 2022-03-15 10:37 maple_hx 阅读(1986) 评论(0) 推荐(0)
摘要: 在pytorch中tensor张量是可以和一个数或者另一个张量进行比较的,但是张量和张量之间进行比较时,需要做到两个张量的纬度相同。比较符号有(>< =等),最后会返回一个与张量纬度大小的张量,张量里面对应的值是一个逻辑数据类型 1 a = torch.Tensor([[1,2],[3,4]]) 2 阅读全文
posted @ 2022-03-14 15:45 maple_hx 阅读(376) 评论(0) 推荐(0)
摘要: 目标检测器中的各种框 anchor:也就是在目标检测训练中,模型所预设的框,在one-stage 的检测器当中,经过backbone提取的特征后,每一个特征内的grid点都会产生若干个anchor,通常为3种或者5种不同形状的预设框 预测框(候选框):anchor经过模型输出微调之后得到的形状大小略 阅读全文
posted @ 2022-02-28 17:12 maple_hx 阅读(580) 评论(0) 推荐(0)
摘要: 在detection2中以及配置且训练好的模型存储在detection2/model_zoo中,其中./config中就存储了相应模型的数据以及配置文件。 一、导入相应的库 1 import cv2 2 import detectron2 3 from detectron2.utils.logger 阅读全文
posted @ 2022-02-23 16:03 maple_hx 阅读(1168) 评论(0) 推荐(0)
摘要: 用代码来举例: 1 def foo(*args, **kwargs): 2 print("args= ", args) 3 print("kwargs= ", kwargs) 4 print(" ") 5 6 if __name__=='__main__': 7 foo(1,2,3,4) 8 foo 阅读全文
posted @ 2022-02-22 15:43 maple_hx 阅读(97) 评论(0) 推荐(0)