dmesquita/azureml-automl-potholes-object-detection

azureml-automl-potholes-object-detection's Issues

ModuleNotFoundError: No module named 'azureml.contrib' while running inference on model.pt for object detection

I have trained yolov5 object detection model with the help of 'AzureML AutoML Potholes detection.ipynb'.
Now I have model.pt and trying to run inference on model.pt but getting error as no module named 'azureml.contrib' though I have installed the module azureml-contrib-automl-dnn-vision.

The code I used:
from azureml.contrib.automl.dnn.vision.common.model_export_utils import load_model, run_inference
from azureml.contrib.automl.dnn.vision.object_detection_yolo.writers.score import _score_with_model

TASK_TYPE = 'image-object-detection'
model_settings = {"img_size": 640, "model_size": "medium", "box_score_thresh": 0.1, "box_iou_thresh": 0.5}

model_wrapper = load_model(TASK_TYPE, 'model.pt', **model_settings)

sample_image = "./img-23_jpg.rf.e6aa0daf83e72ccbf1ea10eb6a6ab3bd.jpg"
with open(sample_image, 'rb') as f:
bytes_img = f.read()

model_response = run_inference(model_wrapper, bytes_img, _score_with_model)

azureml-contrib-automl-dnn-vision package is installed

Error:

Please help me figure this out, Thanks!