06 2021 档案
摘要:本文主要贡献代码模块(文末),在本文中对resnet进行了改写,是一份原始版本模块,里面集成了权重文件pth的载入模块(如函数:init_weights(self, pretrained=None)),layers的冻结模块(如函数:_freeze_stages(self)),更是将其改写成可读性高的代码,若你需要执行该模块,可直接将其代码模块粘贴成.py文件即可。
阅读全文
摘要:目前因项目需要,将检测模型与图像分类结合,完成项目。因此将CBAM模型代码进行整理,仅仅需要train.py与test.py,可分别对图像训练与分类,为了更好学习代码,本文内容分2块,其一将引用 他人博客,简单介绍原理;其二根据改写代码,介绍如何使用,训练自己模型及测试图片。论文:CBAM: Con
阅读全文
摘要:pytorch datasets与dataloader阐释说明,主要简单介绍如何添加自己信息(如 img_meta)作为输入给dataloader,涉及源码更改
阅读全文
摘要:mmdet 一个py搞定自定义测试,并给出结果
阅读全文