How can I use Pytorch/Tensorflow based custom algorithms for ground truth labeling automation?
3 次查看(过去 30 天)
显示 更早的评论
Hi! I am trying to automate ground truth labeling in MATLAB. The default algorithms dont work well for my dataset. ![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/1028105/image.png)
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/1028105/image.png)
Hence, I was looking to use a pytorch based object detection model.
Can someone please guide me to documentation on that?
0 个评论
回答(1 个)
Sivylla Paraskevopoulou
2022-6-11
编辑:Sivylla Paraskevopoulou
2022-6-13
If you want to use a PyTorch model, you can co-execute the object detector in Python. For an example on how to how to automate object labeling in the Image Labeler app using an object detection model trained in Python, see Automate Labeling in Image Labeler Using a Pretrained TensorFlow Object Detector.
Alternatively, you can only import TensorFlow and ONNX models into MATLAB by using the importTensorFlowNetwork and importONNXNetwork functions, respectively. You can convert a PyTorch model into the ONNX model format and then, import the ONNX model into MATLAB using the importONNXNetwork function. For an example on how to import a pretrained ONNX object detector, see Import Pretrained ONNX YOLO v2 Object Detector.
0 个评论
另请参阅
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!