yyc12345/ai-school
1
0
Fork 0
Code Activity

use ignite for exp2

Browse Source
This commit is contained in:
yyc12345
2025-12-02 23:07:27 +08:00
parent 43b807679f
commit 65c56e938c
15 changed files with 246 additions and 794 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
exp2/source/predict.py
View File

@@ -1,3 +1,4 @@
from pathlib import Path
import tensorflow as tf
from PIL import Image
import numpy as np
@@ -33,6 +34,7 @@ class Predict(object):
if __name__ == "__main__":
app = Predict()
app.predict('./test_images/0.png')
app.predict('./test_images/1.png')
app.predict('./test_images/4.png')
images_dir = Path(__file__).resolve().parent.parent / 'test_images'
app.predict(images_dir / '0.png')
app.predict(images_dir / '1.png')
app.predict(images_dir / '4.png')