March 4, 2024, 4:45 p.m. | /u/JustSomeStuffIDid

Computer Vision www.reddit.com

I wrote a guide showing how you can add extra classes to a YOLOv8 pretrained model by adding an extra head. Using this trick, the existing weights and thereby the performance on the pretrained classes are preserved while incurring a little overhead for the extra head that is added.

computervision extra guide head performance pretrained models trick yolov8

Artificial Intelligence – Bioinformatic Expert

@ University of Texas Medical Branch | Galveston, TX

Lead Developer (AI)

@ Cere Network | San Francisco, US

Research Engineer

@ Allora Labs | Remote

Ecosystem Manager

@ Allora Labs | Remote

Founding AI Engineer, Agents

@ Occam AI | New York

AI Engineer Intern, Agents

@ Occam AI | US