Check out Prediktera’s Github repository for examples on how to train your deep learning networks for HSI: https://github.com/Prediktera/Train-Onnx-Image-Segmentation
This user guide shows how to use the object detection available in Breeze. The supported object detection algorithms via ONNX, currently are YOLOv4, YOLOv5 and Faster R-CNN. To utilize this feature add a segmentation node to the Analys Tree, Deep Learning segmentation
Select your pre-trained ONNX model type in the Model Type drop-down and browse to and select the model file, in this case, a Faster R-CNN model file and segmentation.
A Label classification node is automatically added when adding the machine learning segmentation. Add a new line separated class file to the Label node. May be in either .txt or .names files.
Click Apply Changes to evaluate the segmentation, go to the Table tab to see the result.
To train a custom object detection model we recommend using PyTorch and one of the available repositories for training specific version of a network, here is an example which trains a YOLOv5 model with custom data: https://colab.research.google.com/github/Prediktera/Train-Onnx-Image-Segmentation/blob/main/YOLOv5/YOLOv5-colab.ipynb