mmab是fitow自写的完全的openmmlab2.0风格的异常检测代码库,不仅可以实现异常检测模型的训练、评估、测试、可视化、推理。还给未来与其他mm库联动留下想象空间。
mmab完全建议在推荐的docker环境中进行运行!!!
具体参考内部文档
模型名称 | 单卡训练 | 多卡训练 | torch推理 | trt推理 | OPD |
---|---|---|---|---|---|
patchcore | √ | × | √ | √ | × |
模型名称 | 训练集大小 | 分辨率 | 显卡 | 模型种类 | 速度(ms) | 显存占用(MB) |
---|---|---|---|---|---|---|
patchcore | 200(10%) | 256x256 | T4 | trt | 7.5 | 875 |
patchcore | 200(10%) | 256x256 | T4 | torch | 18 | 1571 |
patchcore | 200(10%) | 512x512 | T4 | trt | 49 | 2343 |
patchcore | 200(10%) | 512x512 | T4 | torch | 180 | 6805 |