- 博客(3)
- 收藏
- 关注
原创 视觉异常检测库Anomalib
anomalib提供了多种数据格式:mvtec、btech、visa等。部分数据已有数据集实现类,但项目并不支持。具体支持的数据集格式请全局搜索函数get_datamodule。在src\anomalib\models下有各个算法的实现及该算法的默认运行配置文件config.yaml。配置文件基本都是mvtec的数据格式,可以将数据转换成mvtec数据格式,复制配置文件并修改dataset.path和dataset.category即可。
2024-01-12 19:24:33 1096 1
原创 MMDetection环境配置(Window)
miniconda下载地址:https://docs.conda.io/en/latest/miniconda.html根据提示安装好。
2023-04-02 15:23:30 966 1
原创 MMDetection中对Resnet增加注意力机制Attention的简单方法
以resnet为例子,我在多个尺度的特征层输出增加注意力机制,以此编写一个基类,子类只需要实现这个attention即可。简单的方法是,添加backbone注册修饰器,并在train.py和test.py中,import 该文件。(2)需要注意增加网络的输入输出形状,可能还需要考虑不同宽高输入时网络输出是否与设计一致。(3)增加注意力机制不一定有效涨点,搞数据才是最重要的。(1)这是基于mmdet 2.x版本的改动。
2023-03-25 09:38:28 5872 24
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人