Error when there are more than 100 labels in the model

NickSwardh / YoloDotNet

YoloDotNet - A C# .NET 8.0 project for Classification, Object Detection, OBB Detection, Segmentation and Pose Estimation in both images and videos.

GNU General Public License v3.0

158 stars 28 forks source link

Error when there are more than 100 labels in the model #6

Closed oneprofittips closed 6 months ago

oneprofittips commented 6 months ago

Unhandled exception. System.Exception: There are more labels than available colors. at YoloDotNet.Extensions.OnnxPropertiesExtension.MapLabelsAndColors(String onnxLabelData, ModelType modelType) at YoloDotNet.Extensions.OnnxPropertiesExtension.GetOnnxProperties(InferenceSession session) at YoloDotNet.Data.YoloBase..ctor(String onnxModel, Boolean useCuda, Int32 gpuId) at YoloDotNet.Yolo..ctor(String onnxModel, Boolean cuda, Int32 gpuId)

NickSwardh commented 6 months ago

Thank you for reaching out. I'll have a look at this.

NickSwardh commented 6 months ago

Issue resolved in new version.