search

dhkdnduq / PatchCore_anomaly_detection

Unofficial implementation of PatchCore anomaly detection
Apache License 2.0
25 stars 4 forks source link

Update train.py #1

Closed dhkdnduq closed 3 years ago

dhkdnduq commented 3 years ago

For personal backup 1.Don't use GT 2.Support Libtorch 3.Use all data for heatmap 4.Using ROC Curve for best threshold detection