AAI_2025_Capstone_Chronicles_Combined
14
Draw, Detect, Navigate
Our team’s fully synthetic model, which learned optimal adjustments from previous and
less complicated models, ended up getting a very high testing score of 96.12%. Out of the
1,000 images that contained the doodles which were not used in validation or training, the total
correct doodle predictions for the model was 5,201 out of the 5,411 doodles within the images.
During validation, the model was also benchmarking a 0.6 millisecond pre-process time, 2.4
millisecond inference time, and a 0.6 millisecond post process time. The nano model was able
to achieve these scores with only 25 epochs and a total training time of 0.75 hours. Lastly, and
most importantly, the file size is extremely small coming in at approximately 6.3 MB. Which is
important because our application potentially needs to have the model running locally on the
end user's device, and not have an API endpoint.
Using the exact same model parameters as the YOLOv8 nano, the YOLOv8 small model
performed slightly better than the nano, but took much longer to train, and had a much larger
file size respectively at 22.5 MB. The small model was able to outperform the nano model in
every class for the mAP@0.5:0.95 score by approximately 1%. Although the small model was
able to improve scores significantly, the small model had a longer inference time than the nano,
at nearly three times longer with a speed of 5.9ms. The nano model data ingestion to prediction
which included, preprocess, inference, and postprocess took 2.3 milliseconds total. The only
speed process the small model beat the nano model in was the post process, at 0.4
milliseconds. In times where ever millisecond counts, the nano model achieved the superior but
with the tradeoff of slightly less accurate scores compared to the small. On our groups test
dataset which was one-thousand images, the model was not trained on, or validated with, the
small model was able to get 5241 doodles correctly over, only 40 doodles better than the nano.
41
Made with FlippingBook - Share PDF online