search

xrli-U / MuSc

This is an official PyTorch implementation for "MuSc : Zero-Shot Industrial Anomaly Classification and Segmentation with Mutual Scoring of the Unlabeled Images" (MuSc ICLR2024).
MIT License
247 stars 14 forks source link

Backbone修改 #21

Open XavierWangHX opened 1 month ago

XavierWangHX commented 1 month ago

作者您好！代码默认的配置是'backbone_name': 'ViT-L-14-336', 'feature_layers': [5, 11, 17, 23], 'r_list': [1, 3, 5]， 我将backbone_name改成ViT-B-16后运行报错，我推测是要修改相应的feature_layers或者r_list，可以告诉我怎样修改吗？谢谢！

xrli-U commented 1 month ago

作者您好！代码默认的配置是'backbone_name': 'ViT-L-14-336', 'feature_layers': [5, 11, 17, 23], 'r_list': [1, 3, 5]， 我将backbone_name改成ViT-B-16后运行报错，我推测是要修改相应的feature_layers或者r_list，可以告诉我怎样修改吗？谢谢！

由于ViT-B-16只有12层，所以feature_layers需要改成[2, 5, 7, 11]，在scripts/musc_mvtec_ad.sh脚本中有各种类型backbone的使用配置