去掉了main()中的train()

master
UnknownObject 2 years ago
parent 77f0005b79
commit d13561c25f

@ -51,8 +51,6 @@ def train():
model.export('G:\\Users\\15819\\Desktop\\model01.pkl') # M20.02.pkl
def main():
train()
image_path = 'G:\\Users\\15819\\Desktop\\TrainSet\\SmallCar\\京M88888.jpg'
pred_class, confidence = predict_image(image_path)
print(f"图片类别: {pred_class}, 置信度: {confidence}")

Loading…
Cancel
Save