随笔分类 -  Pytorch/MMDetection

安装MMdetection3D和MinkowskiEngine
摘要：MMdetection3D更新了,只好再次安装,由于CUDA,cuDNN,PyTorch以前已经安装了,这次就不需要安装了,只需要安装MMdetection3D就行了. 1. 安装MMCV 输入以下命令: pip install mmcv-full -f https://download.openm 阅读全文
posted @ 2022-08-14 15:27 MSTK 阅读(1654) 评论(0) 推荐(0) 编辑
MMDetection源码解析：Faster RCNN(8)--BBoxHead类
摘要：BBoxHead类继承自nn.Module类,定义在\mmdet\models\roi_heads\bbox_heads\bbox_head.py中,其作用是输出ROI Pooling的分类和回归值. import torch import torch.nn as nn import torch.n 阅读全文
posted @ 2021-09-24 16:16 MSTK 阅读(1112) 评论(0) 推荐(0) 编辑
MMDetection源码解析：Faster RCNN(7)--ConvFCBBoxHead,Shared2FCBBoxHead和Shared4Conv1FCBBoxHead类
摘要：ConvFCBBoxHead类定义在\mmdet\models\roi_heads\bbox_heads\convfc_bbox_head.py中,其作用是对共享特征层进行卷积和全连接操作,然后在forward到BBoxHead类中,而且也继承自BBoxHead类.convfc_bbox_head. 阅读全文
posted @ 2021-08-09 19:33 MSTK 阅读(955) 评论(0) 推荐(0) 编辑
MMDetection源码解析：Faster RCNN(6)--SingleRoIExtractor类和BaseRoIExtractor类
摘要：SingleRoIExtractor类定义在\mmdet\models\roi_heads\roi_extractors\single_level_roi_extractor.py中,其作用是对ROI特征层进行特征提取,继承自BaseRoIExtractor类. import torch from 阅读全文
posted @ 2021-07-19 15:18 MSTK 阅读(1203) 评论(0) 推荐(0) 编辑
显存不足造成nan错误
摘要：用MMDetection训练过程中,全连接层(Fully Connected Layers,FC)出现nan错误,但是全连接层只是线性组合,而且数值也不大,也没有计算损失函数,怎么可能出现nan错误? 经过研究发现,原来是显存不足造成的,由于显存不足,有些tensor没有了,就造成了nan错误,进行 阅读全文
posted @ 2021-07-08 21:14 MSTK 阅读(1148) 评论(0) 推荐(0) 编辑
MMDetection源码解析：Faster RCNN(5)--TwoStageDetector类
摘要：TwoStageDetector类定义在\mmdet\models\detectors\tew_stage.py中: import torch import torch.nn as nn # from mmdet.core import bbox2result, bbox2roi, build_as 阅读全文
posted @ 2021-06-28 16:11 MSTK 阅读(534) 评论(0) 推荐(0) 编辑
MMDetection源码解析：Faster RCNN(4)--AnchorHead类
摘要：RPNHead继承了AnchorHead(AnchorHead定义在mmdetection/mmdet/models/dense_heads/anchor_head.py文件里面),前面介绍的forward_single(),_get_bboxes_single(),loss()等函数都是重载的An 阅读全文
posted @ 2021-06-10 19:27 MSTK 阅读(686) 评论(0) 推荐(0) 编辑
Permission denied: 'C:\\Users\\Admin\\AppData\\Local\\Temp\\tmp06sdogpt.py'
摘要：运行时出现下面的错误: Traceback (most recent call last): File "D:/Projects/project_name/tools/train.py", line 178, in <module> main() File "D:/Projects/project_ 阅读全文
posted @ 2021-05-01 17:44 MSTK 阅读(2326) 评论(0) 推荐(0) 编辑
CUDA error: device-side assert triggered
摘要：运行时出现下面的错误: C:/cb/pytorch_1000000000000/work/aten/src/ATen/native/cuda/IndexKernel.cu:142: block: [0,0,0], thread: [17,0,0] Assertion `index >= -sizes 阅读全文
posted @ 2021-05-01 17:23 MSTK 阅读(5985) 评论(0) 推荐(0) 编辑
MMDetection源码解析：Faster RCNN(3)--RPN Head类
摘要：Faster RCNN配置文件faster_rcnn_r50_fpn.py中的 backbone=dict( type='ResNet', depth=50, num_stages=4, out_indices=(0, 1, 2, 3), frozen_stages=1, norm_cfg=dict 阅读全文
posted @ 2021-04-14 17:22 MSTK 阅读(1460) 评论(0) 推荐(0) 编辑
MMDetection源码解析：Faster RCNN(2)--FasterRCNN类
摘要：Faster RCNN配置文件faster_rcnn_r50_fpn.py中的 type='FasterRCNN', 说明了这是一个Faster RCNN模型,对应的是mmdetection/mmdet/models/detectors/faster_rcnn.py文件,主要内容如下： from . 阅读全文
posted @ 2021-03-29 17:43 MSTK 阅读(907) 评论(0) 推荐(0) 编辑
MMDetection源码解析：Faster RCNN(1)--配置文件
摘要：faster_rcnn_r50_fpn_1x_coco.py文件位于目录mmdetection/configs/faster_rcnn/下面,主要内容如下: _base_ = [ '../_base_/models/faster_rcnn_r50_fpn.py', '../_base_/datase 阅读全文
posted @ 2021-02-27 23:55 MSTK 阅读(1388) 评论(0) 推荐(0) 编辑
Windows下安装MMDetection
摘要：MMDetection是一个基于Pytorch实现的深度学习和目标检测代码库,包含了Faster-RCNN,YOLO,SSD等主流的目标检测算法代码和已经训练好的模型,方便我们进行目标检测算法的研究.MMDetection的安装步骤如下: 1. 创建一个Conda环境并Activate,很简单,就不 阅读全文
posted @ 2020-12-05 20:27 MSTK 阅读(2079) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示