sjtuplayer / anomalydiffusion

[AAAI 2024] AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model
MIT License
114 stars 14 forks source link

感谢您的开源,我有一个代码问题。 #54

Closed duanyuanyi closed 4 weeks ago

duanyuanyi commented 1 month ago

论文中每类异常生成1000个样本,对应异常定位代码的dataset = MVTec_Anomaly_Detection(args,obj_name,length=500),中的length设为了多少?

sjtuplayer commented 4 weeks ago

论文中每类异常生成1000个样本,对应异常定位代码的dataset = MVTec_Anomaly_Detection(args,obj_name,length=500),中的length设为了多少?

你好,这个length只是用来决定Dataset的size,防止每个epoch的数据过少,所以按照默认设置就好,跟具体数据的数量无关