From b455a5bf3ef623e5cd3346d829c6dc414b7983f5 Mon Sep 17 00:00:00 2001 From: yayoimizuha Date: Sun, 15 Oct 2023 19:16:22 +0900 Subject: [PATCH] update --- .drone.yml | 2 +- matrix_infer.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.drone.yml b/.drone.yml index 2ca7c02..b544777 100644 --- a/.drone.yml +++ b/.drone.yml @@ -24,4 +24,4 @@ steps: - mkdir -p data - $mount_command - ls data/ - - CI=True python finetune/facenet_transfer_learning.py \ No newline at end of file + - # CI=True python finetune/facenet_transfer_learning.py \ No newline at end of file diff --git a/matrix_infer.py b/matrix_infer.py index c7786bd..7181ee9 100644 --- a/matrix_infer.py +++ b/matrix_infer.py @@ -19,7 +19,7 @@ from japanize_matplotlib import japanize device = device('cuda' if is_available() else 'cpu') # device = 'cpu' print(f'device: {device}') -model_path: str = join(datadir(), 'artifact', 'facenet-tl_2023-06-03 23:48:19.808311', 'model.pth') +model_path: str = join(datadir(), 'artifact', 'facenet-tl_2023-10-15 07:08:44.537055', 'model.pth') print(f'model path: {model_path}') input_shape: int = 256 batch_size = 64 @@ -67,3 +67,4 @@ japanize() heatmap_df.max() pyplot.savefig(join(dest_dir, 'confusion_matrix.png')) print(f'acc: {1 - heatmap_df.to_numpy().flatten().sum() / image_folder.__len__()}') +print(image_folder.classes)