|
|
|
@ -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}")
|
|
|
|
|