generated from UnknownObject/PublicRepoDemo
去掉了main()中的train()
This commit is contained in:
@@ -51,8 +51,6 @@ def train():
|
|||||||
model.export('G:\\Users\\15819\\Desktop\\model01.pkl') # M20.02.pkl
|
model.export('G:\\Users\\15819\\Desktop\\model01.pkl') # M20.02.pkl
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
train()
|
|
||||||
|
|
||||||
image_path = 'G:\\Users\\15819\\Desktop\\TrainSet\\SmallCar\\京M88888.jpg'
|
image_path = 'G:\\Users\\15819\\Desktop\\TrainSet\\SmallCar\\京M88888.jpg'
|
||||||
pred_class, confidence = predict_image(image_path)
|
pred_class, confidence = predict_image(image_path)
|
||||||
print(f"图片类别: {pred_class}, 置信度: {confidence}")
|
print(f"图片类别: {pred_class}, 置信度: {confidence}")
|
||||||
|
|||||||
Reference in New Issue
Block a user