missing import seg tutorial (#1235)

sony · Sep 30, 2024 · 3b8f1cc · 3b8f1cc
1 parent 9eb83cb
commit 3b8f1cc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_seg_for_imx500.ipynb b/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_seg_for_imx500.ipynb
@@ -428,6 +428,7 @@
    "source": [
     "from tutorials.mct_model_garden.models_pytorch.yolov8.yolov8 import seg_model_predict\n",
     "from tutorials.mct_model_garden.evaluation_metrics.coco_evaluation import evaluate_yolov8_segmentation\n",
+    "from model_compression_toolkit.core.pytorch.pytorch_device_config import get_working_device\n",
     "device = get_working_device()\n",
     "model = model.to(device)\n",
     "evaluate_yolov8_segmentation(model, seg_model_predict, data_dir='coco', data_type='val2017', img_ids_limit=100, output_file='results.json', iou_thresh=0.7, conf=0.001, max_dets=300,mask_thresh=0.55)"