From 5c4e016cdfed4c19c92e9ef5fd1acd11762aab3d Mon Sep 17 00:00:00 2001
From: pyhqos7bg <wangwei_jmu@163.com>
Date: Thu, 30 May 2024 15:17:46 +0800
Subject: [PATCH] ADD file via upload

---
 model_val.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 model_val.py

diff --git a/model_val.py b/model_val.py
new file mode 100644
index 0000000..0731d9b
--- /dev/null
+++ b/model_val.py
@@ -0,0 +1,27 @@
+from 口罩检测.generate_data import custom_reader
+from 口罩检测.util import train_parameters
+import paddle as paddle
+import paddle.fluid as fluid
+from 口罩检测.VGGNet import VGGNet
+import numpy as np
+eval_reader = paddle.batch(custom_reader(train_parameters['eval_list_path']),
+                            batch_size=train_parameters['train_batch_size'],
+                            drop_last=True)
+with fluid.dygraph.guard():
+    model,_ =fluid.load_dygraph('vgg')
+    vgg =VGGNet()
+    vgg.eval()
+    accs=[]
+    for batch_id,data in enumerate(eval_reader()):
+        x_data = np.array([x[0] for x in data]).astype('float32')
+        y_data = np.array([x[1] for x in data]).astype('int64')
+        y_data = y_data[:,np.newaxis]
+
+        img = fluid.dygraph.to_variable(x_data)
+        label =fluid.dygraph.to_variable(y_data)
+
+        out,acc= vgg(img,label)
+
+        label = np.argmax(out.numpy())
+        accs.append(acc.numpy()[0])
+        print(np.mean(accs))
\ No newline at end of file