remove onnx
Browse files
app.py
CHANGED
|
@@ -372,6 +372,8 @@ def find_angles(image,order):
|
|
| 372 |
y_pred = model[i].predict(x)
|
| 373 |
result_preds.append(y_pred.round(2))
|
| 374 |
return result_preds # yxz-orderd x,y,z
|
|
|
|
|
|
|
| 375 |
import onnxruntime as ort
|
| 376 |
def estimate3(model_key,features_values):
|
| 377 |
model_path=f"models/{model_key}.onnx"
|
|
@@ -385,6 +387,7 @@ def find_angles(image,order):
|
|
| 385 |
result_preds = ort_session.run(None, {input_name: input_data})
|
| 386 |
#print((result_preds))
|
| 387 |
return result_preds[0] # yxz-orderd x,y,z
|
|
|
|
| 388 |
|
| 389 |
|
| 390 |
#short_result = estimate('linear-svr-xyz_5.joblib','linear-svr-xyz_5_scaler.joblib',features_values)
|
|
|
|
| 372 |
y_pred = model[i].predict(x)
|
| 373 |
result_preds.append(y_pred.round(2))
|
| 374 |
return result_preds # yxz-orderd x,y,z
|
| 375 |
+
|
| 376 |
+
"""
|
| 377 |
import onnxruntime as ort
|
| 378 |
def estimate3(model_key,features_values):
|
| 379 |
model_path=f"models/{model_key}.onnx"
|
|
|
|
| 387 |
result_preds = ort_session.run(None, {input_name: input_data})
|
| 388 |
#print((result_preds))
|
| 389 |
return result_preds[0] # yxz-orderd x,y,z
|
| 390 |
+
"""
|
| 391 |
|
| 392 |
|
| 393 |
#short_result = estimate('linear-svr-xyz_5.joblib','linear-svr-xyz_5_scaler.joblib',features_values)
|