GitHub/test_nine
1
0
Fork 0
mirror of https://github.com/luguoyixiazi/test_nine.git synced 2025-12-06 14:52:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update predict.py

Browse Source
This commit is contained in:
luguoyixiazi
2025-09-06 20:29:18 +08:00
committed by GitHub
parent f3d2951f80
commit c26e354c2f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
predict.py
View File

@@ -205,7 +205,7 @@ def predict_onnx_pdl(images_path):
print(f"识别完成{answer}，耗时: {time.time() - start}") print(f"识别完成{answer}，耗时: {time.time() - start}")
with open(os.path.join(images_path,"nine.jpg"),'rb') as f: with open(os.path.join(images_path,"nine.jpg"),'rb') as f:
bg_image = f.read() bg_image = f.read()
draw_points_on_image(bg_image, answer) # draw_points_on_image(bg_image, answer)
return answer return answer
def calculate_iou(boxA, boxB): def calculate_iou(boxA, boxB):
@@ -369,3 +369,4 @@ if __name__ == "__main__":
# 使用PP-HGNetV2-B4.onnx # 使用PP-HGNetV2-B4.onnx
#predict_onnx_pdl(r'img_saved\img_fail\7fe559a85bac4c03bc6ea7b2e85325bf') #predict_onnx_pdl(r'img_saved\img_fail\7fe559a85bac4c03bc6ea7b2e85325bf')
predict_onnx_dfine(r"n:\爬点选\dataset\3f98ff0c91dd4882a8a24d451283ad96.jpg",True) predict_onnx_dfine(r"n:\爬点选\dataset\3f98ff0c91dd4882a8a24d451283ad96.jpg",True)