mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add label_list file

pull/1599/head
weishengyu 2022-01-25 15:33:44 +08:00
parent 377a7b4b2f
commit 88e6387b2c
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ppcls/utils/gallery2fc.py
View File

@ -53,6 +53,12 @@ class GalleryLayer(paddle.nn.Layer):
gallery_labels.append(line[1].strip()) gallery_labels.append(line[1].strip())
self.gallery_layer = paddle.nn.Linear(embedding_size, len(self.gallery_images), bias_attr=False) self.gallery_layer = paddle.nn.Linear(embedding_size, len(self.gallery_images), bias_attr=False)
self.gallery_layer.skip_quant = True self.gallery_layer.skip_quant = True
output_label_str = ""
for i, label_i in enumerate(gallery_labels):
output_label_str += "{} {}\n".format(i, label_i)
output_path = configs["Global"]["save_inference_dir"] + "_label.txt"
with open(output_path, "w") as f:
f.write(output_label_str)
def forward(self, x, label=None): def forward(self, x, label=None):
x = paddle.nn.functional.normalize(x) x = paddle.nn.functional.normalize(x)