Add full yolov8 compatible post-processing (#115)

This updates the local detection to yolov8-based models from the older yolov5 based models.

This includes significant additional post-processing of the raw yolov8 output, but also means that the smaller nano yolo model is available as well as all of the updated ALVINN thorax training data.

Signed-off-by: Justin Georgi <justin.georgi@gmail.com>

Reviewed-on: Georgi_Lab/ALVINN_f7#115
This commit is contained in:
2024-03-04 15:03:20 -07:00
parent 7f1aa63f07
commit 069ad74e30
24 changed files with 252 additions and 42 deletions

View File

@@ -289,7 +289,7 @@
import submitMixin from './submit-mixin'
import detectMixin from './local-detect'
import thoraxClasses from '../models/thorax_tfwm/classes.json'
import thoraxClasses from '../models/thorax-0.1.0-n960/classes.json'
export default {
mixins: [submitMixin, detectMixin],
@@ -334,9 +334,9 @@
this.detectorName = 'thorax'
this.classesList = thoraxClasses
/* VITE setting */
this.modelLocation = '../models/thorax_tfwm/model.json'
this.modelLocation = '../models/thorax-0.1.0-n960/model.json'
/* PWA Build setting */
//this.modelLocation = './models/thorax_tfwm/model.json'
//this.modelLocation = './models/thorax-0.1.0-n960/model.json'
this.modelLocationCordova = 'https://localhost/models/thorax_tfwm/model.json'
break;
case 'abdomen':