search

sony / model_optimization

Model Compression Toolkit (MCT) is an open source project for neural network model optimization under efficient, constrained hardware. This project provides researchers, developers, and engineers advanced quantization and compression tools for deploying state-of-the-art neural networks.
https://sony.github.io/model_optimization/
Apache License 2.0
307 stars 49 forks source link

ModuleNotFoundError: No module named 'sony_custom_layers.pytorch.object_detection' #1218

Closed alexander-sony closed 3 weeks ago

alexander-sony commented 3 weeks ago

Issue Type

Bug

Source

pip (mct-nightly)

MCT Version

2.2.0.20240916.525

OS Platform and Distribution

google colab

Python version

3.10.12

Describe the issue

error running tutorial in colab https://github.com/sony/model_optimization/blob/main/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_for_imx500.ipynb

At model creation step:

from tutorials.mct_model_garden.models_pytorch.yolov8.yolov8 import ModelPyTorch, yaml_load, model_predict
cfg_dict = yaml_load("tutorials/mct_model_garden/models_pytorch/yolov8/yolov8n.yaml", append_filename=True)  # model dict
model = ModelPyTorch.from_pretrained("SSI-DNN/pytorch_yolov8n_detection_640x640", cfg=cfg_dict)
model.eval()

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
[<ipython-input-5-7698929858e2>](https://localhost:8080/#) in <cell line: 1>()
----> 1 from tutorials.mct_model_garden.models_pytorch.yolov8.yolov8 import ModelPyTorch, yaml_load, model_predict
      2 cfg_dict = yaml_load("tutorials/mct_model_garden/models_pytorch/yolov8/yolov8n.yaml", append_filename=True)  # model dict
      3 model = ModelPyTorch.from_pretrained("SSI-DNN/pytorch_yolov8n_detection_640x640", cfg=cfg_dict)
      4 model.eval()

[/content/tutorials/mct_model_garden/models_pytorch/yolov8/yolov8.py](https://localhost:8080/#) in <module>
     51 from tutorials.mct_model_garden.models_pytorch.yolov8.yolov8_postprocess import postprocess_yolov8_keypoints
     52 if importlib.util.find_spec("sony_custom_layers"):
---> 53     from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms
     54 
     55 

ModuleNotFoundError: No module named 'sony_custom_layers.pytorch.object_detection'

Observe:

pip show sony-custom-layers-dev
Name: sony-custom-layers-dev
Version: 0.3.0.dev1
Summary: Sony Custom Layers package
...

Expected behaviour

No response

Code to reproduce the issue

see above

Log output

No response

alexander-sony commented 3 weeks ago

see also pytorch_yolov8n_pose_for_imx500.ipynb

Idan-BenAmi commented 3 weeks ago

Hi @alexander-sony , Thanks for your feedback, we fixed wrong version of sony-custom-layers in #1219