[mod] percent

app.py

@@ -36,7 +36,9 @@ if st.button("不審者を検知"):
     st.title("解析結果")
 
     for a,b in zip(labels, scores):
-        st.write(a,b)
+        prob = float(b) * 100
+        st.write(a)
+        st.write(f"AIの確信度： {prob} %")
 
     if len(indices) != 0:
         st.warning('不審者が検知された可能性があります', icon="⚠️")

run.py

@@ -61,12 +61,14 @@ def inference(image_pil):
     num_classes = 5
     backbone = resnet_fpn_backbone('resnet18', False)
     model = FasterRCNN(backbone, num_classes)
-    model.eval()
     state_dict = torch.load('model/model/densenet-model-9-mAp--1.0.pth',map_location=device)
     model.load_state_dict(state_dict["model"])
+    
+    model.eval()
     _transform = T.Compose([T.ToTensor()])
     image = image_pil.convert("RGB")
     image = _transform(image)
-    output = model([image])
+    with torch.no_grad():
+        output = model([image])
     res = postprocess(image_pil, output)
     return output, res