laitathei / YOLOv8-ONNX-RKNN-HORIZON-TensorRT-Segmentation

Inference YOLOv8 segmentation on ONNX, RKNN, Horizon and TensorRT
MIT License
30 stars 1 forks source link

Training the yolov8 neural network #2

Open lera5648 opened 6 months ago

lera5648 commented 6 months ago

Good afternoon I tried converting rknn to a .pt model trained on the original ultralitics repository. But after converting the model to rknn, there were a large number of false and incorrect predictions (this was observed on the rockchip board. On the intel processor everything worked fine). Tell me, is there any special technique for training yolov8, in which the model is successfully transformed?

laitathei commented 6 months ago

In the Intel processor, you can only get the simulation result, which can be totally different from the rockchip result When you tried to implement your rknn model into rockchip, you need to consider the which model of rockchip and the post-process program carefully