ImportError: cannot import name 'DSegment' from 'models.yolo' (/kaggle/working/yolov9/models/yolo.py)`

[dsbyprateekg]

Hi,

I have got below error while converting custom trained model to onnx-

Traceback (most recent call last): File "/kaggle/working/yolov9/export_onnxtrt.py", line 19, in <module> from models.yolo import ClassificationModel, Detect, DDetect, DualDetect, DualDDetect, DetectionModel, SegmentationModel, DSegment ImportError: cannot import name 'DSegment' from 'models.yolo' (/kaggle/working/yolov9/models/yolo.py)

My CUDA Version: 12.6
Python: 3.10.14
Torch: 2.4.0