fixed classifier and added in a preliminary results view that shows what pokemon are currently being looked at.

This commit is contained in:
Lucas Oskorep
2022-06-22 21:44:15 -04:00
parent ebfbfb503d
commit 9ec737db46
10 changed files with 305 additions and 355 deletions
+21
View File
@@ -0,0 +1,21 @@
/// Represents the recognition output from the model
class Recognition {
/// Index of the result
final int _id;
/// Label of the result
final String _label;
/// Confidence [0.0, 1.0]
final double _score;
Recognition(this._id, this._label, this._score);
int get id => _id;
String get label => _label;
double get score => _score;
@override
String toString() {
return 'Recognition(id: $id, label: $label, score: $score)';
}
}
+18
View File
@@ -0,0 +1,18 @@
class Stats {
int totalTime;
int preProcessingTime;
int inferenceTime;
int postProcessingTime;
Stats(
{this.totalTime = -1,
this.preProcessingTime = -1,
this.inferenceTime = -1,
this.postProcessingTime = -1});
@override
String toString() {
return 'Stats{totalPredictTime: $totalTime, preProcessingTime: $preProcessingTime, inferenceTime: $inferenceTime, postProcessingTime: $postProcessingTime}';
}
}