- 博客(5)
- 收藏
- 关注
原创 detctron2 rpn 损失函数解读
一. 数据准备工作detectron2.modeling.proposal_generator.rpn.RPN.label_and_sample_anchors重点代码行解读match_quality_matrix = retry_if_cuda_oom(pairwise_iou)(gt_boxes_i, anchors)match_quality_matrix:代表了每个label 对应的所有 anchor 的 IOU 矩阵比如我们给某张图片进行标记了 100个 label ,然后我们为这个
2021-02-27 16:29:53
1843
2
原创 detectron 源码解读-巧妙计算 anchor 和 gt_boxes 相交面积
在faster-rcnn 中涉及了 IOU 的计算,而 IOU 就是要先计算相交面积。在detectron 的源码中,仅仅用了几行代码就实现了 anchor 集合 和 gt_boxes 集合的相交面积矩阵计算代码位置如下# boxes1 代表的是 gt_boxes # boxes2 代表的是 系统生成的 上万个 anchor def pairwise_intersection(boxes1: Boxes, boxes2: Boxes) -> torch.Tensor: """
2021-02-25 16:18:42
913
原创 ubuntu CUDA卸载重装
sudo apt remove nvidia*sudo apt remove cuda*sudo apt remove cudnn*如果之前是 deb 包安装的,还要操作如下步骤:sudo apt-key list | grep cudapub rsa4096 2016-06-24 [SC] AE09 FE4B BD22 3A84 B2CC FCE3 F60F 4B3D 7FA2 AF80uid [ 未知 ] cudatools <cudatool
2021-02-25 16:01:35
1583
原创 detectron ResizeShortestEdge 的实现探究
最近在做项目的时候,发现训练完了之后,输入不同尺寸的图,精准度会不同。于是仔细的看了下对于不同尺寸的增广训练。训练阶段用到的参数 :# 后面跟的值是举例值,使用者可根据自己的图片大小进行调整 MAX_SIZE_TRAIN: 1333 # 图片的长边 ,必须小于 1333。 第一优先级限制 MIN_SIZE_TRAIN: (640 , 800) #图片的长边,会缩放到这个尺寸,在满足 MAX_SIZE_TRAIN 的前提下,才会满足这个条件举
2021-02-18 18:16:00
1639
2
转载 【转载】Ubuntu samba 共享给windows
https://websiteforstudents.com/install-samba-on-ubuntu-20-04-18-04/
2021-01-19 15:22:46
135
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人