tomokazu
/
helloproject-ai
mirror of https://github.com/yayoimizuha/helloproject-ai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update
continuous-integration/drone/push Build is passing Details

This commit is contained in:
yayoimizuha 2023-10-15 19:16:22 +09:00
parent 49ab75dc83
commit b455a5bf3e
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.drone.yml
View File

@ -24,4 +24,4 @@ steps:
- mkdir -p data - mkdir -p data
- $mount_command - $mount_command
- ls data/ - ls data/
- CI=True python finetune/facenet_transfer_learning.py - # CI=True python finetune/facenet_transfer_learning.py

3
matrix_infer.py
View File

@ -19,7 +19,7 @@ from japanize_matplotlib import japanize
device = device('cuda' if is_available() else 'cpu') device = device('cuda' if is_available() else 'cpu')
# device = 'cpu' # device = 'cpu'
print(f'device: {device}') 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}') print(f'model path: {model_path}')
input_shape: int = 256 input_shape: int = 256
batch_size = 64 batch_size = 64
@ -67,3 +67,4 @@ japanize()
heatmap_df.max() heatmap_df.max()
pyplot.savefig(join(dest_dir, 'confusion_matrix.png')) pyplot.savefig(join(dest_dir, 'confusion_matrix.png'))
print(f'acc: {1 - heatmap_df.to_numpy().flatten().sum() / image_folder.__len__()}') print(f'acc: {1 - heatmap_df.to_numpy().flatten().sum() / image_folder.__len__()}')
print(image_folder.classes)