1
This commit is contained in:
@@ -61,7 +61,14 @@ class DetectorWorker {
|
||||
try {
|
||||
if (models == null || models.isEmpty) return null;
|
||||
final payload = <List<Object?>>[
|
||||
for (final m in models) [m.bytes, m.labels, m.datasetId, m.datasetName],
|
||||
for (final m in models)
|
||||
[
|
||||
m.bytes,
|
||||
m.labels,
|
||||
m.datasetId,
|
||||
m.datasetName,
|
||||
m.variant,
|
||||
],
|
||||
];
|
||||
|
||||
final responses = ReceivePort();
|
||||
@@ -222,11 +229,15 @@ Future<void> _workerMain(SendPort mainPort) async {
|
||||
for (final entry in list[1] as List) {
|
||||
final e = entry as List;
|
||||
final name = e.length > 3 ? e[3] as String : '';
|
||||
// 模型名带档位标识(数据集名+档位,框来源可辨 s/n)
|
||||
final variant = e.length > 4 ? e[4] as String : '';
|
||||
final displayName =
|
||||
variant.isEmpty ? name : '$name($variant)';
|
||||
final d = await TfliteDetector.fromBuffer(
|
||||
e[0] as Uint8List,
|
||||
(e[1] as List).cast<String>(),
|
||||
modelId: (e[2] as num).toInt(),
|
||||
modelName: name,
|
||||
modelName: displayName,
|
||||
);
|
||||
if (d == null) {
|
||||
failures.add(name.isEmpty ? 'unknown' : name);
|
||||
@@ -369,7 +380,7 @@ Future<void> _workerMain(SendPort mainPort) async {
|
||||
results.addAll(dets);
|
||||
}
|
||||
results = mergeAcrossModels(results, TfliteDetector.iouThreshold);
|
||||
// 低分环颈雉鸡框过视觉先验(颜色/位置),减少户外误报
|
||||
// 低分目标框过视觉先验(颜色/位置),减少户外误报
|
||||
results = VisualPrior.filter(
|
||||
results,
|
||||
planes: planes,
|
||||
|
||||
Reference in New Issue
Block a user