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

请问合成的异常数据,是否用于有监督训练呢? #4

Closed Justhuayu closed 3 months ago

Justhuayu commented 3 months ago

我在你提供的数据集里面,并没有找到对应的标签。是训练时直接生成标签,还是说进行无监督训练呢?

cnulab commented 3 months ago

RealNet是一个自监督模型,在训练时通过图像混合任务来生成标签(Mask)。

Justhuayu commented 3 months ago

请问生成标签的代码在哪个位置呢?刚接触深度学习,看不太懂

cnulab commented 3 months ago

在datasets/realnet_dataset.py文件中

Justhuayu commented 3 months ago

明白了,十分感谢!!