Visualize weights in pytorch

First layer convolutional features

其实CV模型的第一层卷积

1.可以直接打印核
2.是不是代表了模型的某种特征?
在这里插入图片描述
上图

经典alexnet
在这里插入图片描述

from torchvision import utils
import torchvision.transforms as T
from PIL import Image

## load your network as net

grid = utils.make_grid(net.{conv}.weight, nrow=8, normalize=True, padding=1)
save=T.Compose([T.ToPILImage()])
save(grid).save('viswight.png')

resnet50 在danbooru上训
在这里插入图片描述
inception做特征抽取的bbox动漫头像回归检测模型
在这里插入图片描述
试试stylegan的torgb层?
是不是可以torgb的k大一点?

PCARN动漫模型
在这里插入图片描述

我错了
应该这么看
不是什么RGB
RGB更离谱,看不出规律
在这里插入图片描述

再发现

这是conv的bias

死滤波器
在这里插入图片描述
这啥玩意这么高的bias???
在这里插入图片描述
我开始怀疑了
是不是滤波器真的能打印
norm,clip到255做灰度,很多细节可能都丢了

或许这玩意就像暗能量一样(字面意思),大部分信息都在细节里,编码方式无法看懂,常规能量就是那看得到的规律

又一次,人类经验,先入为主,不可靠,,

### PyTorch Mask R-CNN 实现教程 #### 安装环境配置 为了成功运行 `Mask_RCNN_Pytorch` 项目,需先设置合适的开发环境。建议创建一个新的 Python 虚拟环境,并安装指定版本的依赖库[^2]。 ```bash conda create -n maskrcnn python=3.7 conda activate maskrcnn pip install torch==0.4.0 torchvision numpy opencv-python h5py matplotlib scikit-image ``` #### 获取源码与预训练权重文件 访问该项目托管平台获取最新版代码仓库: ```bash git clone https://gitcode.com/gh_mirrors/ma/Mask_RCNN_Pytorch.git cd Mask_RCNN_Pytorch wget http://images.cocodataset.org/zips/train2017.zip unzip train2017.zip rm train2017.zip ``` 下载官方提供的 COCO 预训练模型参数用于迁移学习加速收敛过程: ```bash mkdir logs/ wget -P ./logs/ http://download.tensorflow.org/models/object_detection/mask_rcnn_inception_v2_coco_2018_01_28.tar.gz tar -xzvf ./logs/mask_rcnn_inception_v2_coco_2018_01_28.tar.gz -C ./logs/ rm ./logs/mask_rcnn_inception_v2_coco_2018_01_28.tar.gz ``` #### 使用示例脚本检测单张图片中的物体实例边界框及掩膜区域 加载已保存好的网络结构及其对应的权值参数后即可调用预测接口处理输入图像得到输出结果[^1]: ```python from mrcnn import model as modellib, utils import os import cv2 import random import visualize class InferenceConfig(Config): NAME = "coco" GPU_COUNT = 1 IMAGES_PER_GPU = 1 NUM_CLASSES = 1 + 80 # Background + coco classes if __name__ == '__main__': ROOT_DIR = os.getcwd() MODEL_DIR = os.path.join(ROOT_DIR, "logs") config = InferenceConfig() # Recreate the model in inference mode model = modellib.MaskRCNN(mode="inference", config=config, model_dir=MODEL_DIR) # Load weights trained on MS-COCO model.load_weights(os.path.join(MODEL_DIR, 'mask_rcnn_coco.h5'), by_name=True) image_path = './test.jpg' img = cv2.imread(image_path) results = model.detect([img], verbose=1)[0] r = results[0] ax = plt.gca() visualize.display_instances(img, r['rois'], r['masks'], r['class_ids'], ['BG', ...], r['scores'], ax=ax, title="Predictions") fig.canvas.print_png('output.png') ``` 此段程序展示了如何利用预训练过的 Mask R-CNN 对新采集到的数据集执行推理操作并绘制出目标对象的位置以及形状轮廓图。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值