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:
@@ -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':
|
||||
|
||||
Reference in New Issue
Block a user