okzhili / Cartoon-face-recognition

CCF-BDCI 训练赛 漫画照片人脸识别TOP1方案
21 stars 5 forks source link

找不到模型文件 #9

Open superfl8 opened 1 year ago

superfl8 commented 1 year ago

请问博主配置文件中的模型PRETRAIN_PATH: '/home/resnest269-0cc87c48.pth'从哪下载,以及WEIGHT: '/data/final/resnest269_256/16/resnest269_40.pth'也没有找到

okzhili commented 1 year ago

请从这里下载https://github.com/zhanghang1989/ResNeSt, /data/final/resnest269_256/16/resnest269_40.pth是训练得到的,你可以把这里替换为你训练得到的文件。

superfl8 commented 1 year ago

十分感谢您的回复! 我们在复现该项目时还发现在./datasets/dm.py 中 line 61 报错,是由于 FR_Probe_C2P.txt 文件中有的人名只有一个单词,导致列数从3列变成了2列,于是在query = np.loadtxt(...)中增加了一个分隔符delimiter='\t',接着在./datasets/base.py 中line 39 报错,其原因是cams是不可哈希化的(unhashable type: 'numpy.ndarray'),cams中的元素如下所示: ... array(['John Travolta', 'C00034'], dtype='<U24'), array(['John Travolta', 'C00035'], dtype='<U24'), array(['John Travolta', 'C00036'], ... 请问这个问题该如何解决?再次感谢您的答复!