cnulab / RealNet

Offical implementation of "RealNet: A Feature Selection Network with Realistic Synthetic Anomaly for Anomaly Detection (CVPR 2024)"
MIT License
143 stars 5 forks source link

是否为一个类别训练一个检测模型 #17

Open truefaker opened 2 months ago

truefaker commented 2 months ago

请问,是为每个类别单独训练一个检测模型么?

cnulab commented 2 months ago

是的,您可以参考https://github.com/cnulab/RealNet/issues/6

truefaker commented 2 months ago

感谢回复,请问您是否尝试过如UniAD一样的多类别异常检测设置呢,即一个数据集一个检测模型

cnulab commented 2 months ago

是的,在论文附录的C.5节我们讨论了RealNet在多个类别下的设置。

truefaker commented 2 months ago

请问您是否测试过多类设置下PRO指标,以及visa数据集上的表现

cnulab commented 2 months ago

非常抱歉,我没有这些数据😞 如果你想在多类的设置下测试RealNet的表现,你需要对该仓库的代码进行一些修改,以下是一些要点:

  1. 在realnet.yaml中配置使用DTD进行异常合成
  2. 训练和测试文件使用train_uni.json和test_uni.json
  3. 重建网络结构使用models.simple_recon.SimpleReconstructionLayer
  4. 在异常合成过程中禁用前景分割

感谢您对我们工作的关注,祝您生活愉快!

truefaker commented 2 months ago

感谢您的回复