Skip to content

Step by step train and export customized yolo (Checking the changes made to the model. Export for android application)

Notifications You must be signed in to change notification settings

Gainward777/yolov5_custom_training

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

Training, export and test Customized YOLOv5
Dataset NUMBERS Computer Vision
The result can be seen: Android App

The dataset used in this work did not contain small numbers usually denoting parts of the whole.
To isolate them from the total mass, you can use the calculation of the area of the predicted bonding boxes.
To do this, in the new_sorter function of the model, you need to add something like that:

x1 = non_max[:, 0]
y1 = non_max[:, 1]
x2 = non_max[:, 2]
y2 = non_max[:, 3] 

areas = (x2 - x1) * (y2 - y1)

Further, you can work with the corresponding values both in the function itself and on the device.

About

Step by step train and export customized yolo (Checking the changes made to the model. Export for android application)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published