easy-Fpn源码解读(五):rpn

2024-04-02 05:08
文章标签 源码 解读 easy rpn fpn

本文主要是介绍easy-Fpn源码解读(五):rpn,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

目录

  • easy-Fpn源码解读(五):rpn
    • region_proposal_network.py代码解析

easy-Fpn源码解读(五):rpn

region_proposal_network.py代码解析

from typing import Tuple, Listimport numpy as np
import torch
from torch import nn, Tensor
from torch.nn import functional as Ffrom bbox import BBox
from nms.nms import NMSclass RegionProposalNetwork(nn.Module):def __init__(self, num_features_out: int, anchor_ratios: List[Tuple[int, int]], anchor_scales: List[int], pre_nms_top_n: int, post_nms_top_n: int):super().__init__()self._features = nn.Sequential(nn.Conv2d(in_channels=num_features_out, out_channels=512, kernel_size=3, padding=1),nn.ReLU())# 通道数从256维升至512维self._anchor_ratios = anchor_ratiosself._anchor_scales = anchor_scalesnum_anchor_ratios = len(self._anchor_ratios)num_anchor_scales = len(self._anchor_scales)num_anchors = num_anchor_ratios * num_anchor_scalesself._pre_nms_top_n = pre_nms_top_nself._post_nms_top_n = post_nms_top_nself._objectness = nn.Conv2d(in_channels=512, out_channels=num_anchors * 2, kernel_size=1)self._transformer = nn.Conv2d(in_channels=512, out_channels=num_anchors * 4, kernel_size=1)# 因为是1*1的kernel,所以不会改变特征图的size,# 对一个特征图像上的每一个点来说,会生成num*anchors个anchor,而每个anchor都有两个前景/背景概率# 和4个坐标位置,因此对于self._objectness来说两个通道对应一个anchor,而对于self._transformer来说# 4个通道对应一个anchor。这4个通道是一个anchor的四个位置的修正偏移量。# 所以在forward的里,才会将通道数换至最后一维,再展开。这样每一行就会对应一个anchordef forward(self, features: Tensor, image_width: int, image_height: int) -> Tuple[Tensor, Tensor]:features = self._features(features)objectnesses = self._objectness(features)  # anchor前景/背景修正transformers = self._transformer(features)

这篇关于easy-Fpn源码解读(五):rpn的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!


原文地址:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.chinasem.cn/article/869045

相关文章

解读@ConfigurationProperties和@value的区别

《解读@ConfigurationProperties和@value的区别》:本文主要介绍@ConfigurationProperties和@value的区别及说明,具有很好的参考价值,希望对大家... 目录1. 功能对比2. 使用场景对比@ConfigurationProperties@Value3. 核

8种快速易用的Python Matplotlib数据可视化方法汇总(附源码)

《8种快速易用的PythonMatplotlib数据可视化方法汇总(附源码)》你是否曾经面对一堆复杂的数据,却不知道如何让它们变得直观易懂?别慌,Python的Matplotlib库是你数据可视化的... 目录引言1. 折线图(Line Plot)——趋势分析2. 柱状图(Bar Chart)——对比分析3

Jupyter notebook安装步骤解读

《Jupyternotebook安装步骤解读》:本文主要介绍Jupyternotebook安装步骤,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、开始安装二、更改打开文件位置和快捷启动方式总结在安装Jupyter notebook 之前,确认您已安装pytho

Java中的StringUtils.isBlank()方法解读

《Java中的StringUtils.isBlank()方法解读》:本文主要介绍Java中的StringUtils.isBlank()方法,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑... 目录所在库及依赖引入方法签名方法功能示例代码代码解释与其他方法的对比总结StringUtils.isBl

对Django中时区的解读

《对Django中时区的解读》:本文主要介绍对Django中时区的解读方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录背景前端数据库中存储接口返回AI的解释问题:这样设置的作用答案获取当前时间(自动带时区)转换为北京时间显示总结背景设置时区为北京时间 TIM

Java中的内部类和常用类用法解读

《Java中的内部类和常用类用法解读》:本文主要介绍Java中的内部类和常用类用法,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录内部类和常用类内部类成员内部类静态内部类局部内部类匿名内部类常用类Object类包装类String类StringBuffer和Stri

使用easy connect之后,maven无法使用,原来需要配置-Djava.net.preferIPv4Stack=true问题

《使用easyconnect之后,maven无法使用,原来需要配置-Djava.net.preferIPv4Stack=true问题》:本文主要介绍使用easyconnect之后,maven无法... 目录使用easGWowCy connect之后,maven无法使用,原来需要配置-DJava.net.pr

JVM垃圾回收机制之GC解读

《JVM垃圾回收机制之GC解读》:本文主要介绍JVM垃圾回收机制之GC,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、死亡对象的判断算法1.1 引用计数算法1.2 可达性分析算法二、垃圾回收算法2.1 标记-清除算法2.2 复制算法2.3 标记-整理算法2.4

Android实现一键录屏功能(附源码)

《Android实现一键录屏功能(附源码)》在Android5.0及以上版本,系统提供了MediaProjectionAPI,允许应用在用户授权下录制屏幕内容并输出到视频文件,所以本文将基于此实现一个... 目录一、项目介绍二、相关技术与原理三、系统权限与用户授权四、项目架构与流程五、环境配置与依赖六、完整

Android实现定时任务的几种方式汇总(附源码)

《Android实现定时任务的几种方式汇总(附源码)》在Android应用中,定时任务(ScheduledTask)的需求几乎无处不在:从定时刷新数据、定时备份、定时推送通知,到夜间静默下载、循环执行... 目录一、项目介绍1. 背景与意义二、相关基础知识与系统约束三、方案一:Handler.postDel