1
This commit is contained in:
@@ -135,6 +135,26 @@ void main() {
|
||||
expect(m2.models.first.version, 'v2.0.0');
|
||||
});
|
||||
|
||||
test('autoUpdate:已激活模型出新版本,refresh 后自动重下并立即生效', () async {
|
||||
final m = manager(client([_item()]));
|
||||
await m.refresh();
|
||||
await m.downloadModel(m.catalog.first);
|
||||
expect(downloadHits, 1);
|
||||
|
||||
// 重启 + 服务器目录出 v2:refresh 内部 autoUpdate 自动重下(无需手动)
|
||||
final m2 = manager(client([_item(version: 'v2.0.0')]));
|
||||
await m2.refresh();
|
||||
for (var i = 0; i < 50 && downloadHits < 2; i++) {
|
||||
await Future<void>.delayed(const Duration(milliseconds: 20));
|
||||
}
|
||||
|
||||
expect(downloadHits, 2, reason: '新版本应自动重下');
|
||||
expect(m2.isActive(7), isTrue, reason: '自动更新应保持激活');
|
||||
expect(m2.models.first.version, 'v2.0.0',
|
||||
reason: '自动更新后立即生效新版本字节');
|
||||
expect(m2.models.first.bytes, _modelBytes);
|
||||
});
|
||||
|
||||
test('sha256 不匹配:重试后失败、不激活、错误可见', () async {
|
||||
final m = manager(client([_item()]));
|
||||
await m.refresh();
|
||||
|
||||
@@ -28,12 +28,23 @@ void main() {
|
||||
expect(merged.first.modelName, '野兔模型');
|
||||
});
|
||||
|
||||
test('不同类别互不压制', () {
|
||||
test('不同类别重叠:去重取高分(实测多模型对同一目标检异类别)', () {
|
||||
final all = [
|
||||
box('pheasant', 0.3, 0.5, 0.5, modelId: 1),
|
||||
box('hare', 0.7, 0.5, 0.5, modelId: 2), // 同位置但不同类别
|
||||
];
|
||||
final merged = mergeAcrossModels(all, 0.45);
|
||||
expect(merged.length, 1);
|
||||
expect(merged.first.label, 'hare');
|
||||
expect(merged.first.score, 0.7);
|
||||
});
|
||||
|
||||
test('不同类别不重叠:都保留', () {
|
||||
final all = [
|
||||
box('pheasant', 0.3, 0.1, 0.1, modelId: 1),
|
||||
box('hare', 0.7, 0.8, 0.8, modelId: 2), // 远处互不重叠
|
||||
];
|
||||
final merged = mergeAcrossModels(all, 0.45);
|
||||
expect(merged.length, 2);
|
||||
});
|
||||
|
||||
|
||||
@@ -30,6 +30,25 @@ void main() {
|
||||
expect(result[0].score, greaterThan(0.5));
|
||||
});
|
||||
|
||||
test('bigBox_containsSmallBox_deduped', () {
|
||||
// 同目标一大一小两框:标准 IoU = 0.04/0.13 ≈ 0.31 < 0.45 会漏,
|
||||
// minIoU = inter/小框 = 1 必须去重(2026-09-01 用户实测:单模型也有重叠框)
|
||||
final a = box(0.3, 0.3, 0.5, 0.5, 0.8); // 小框 0.2x0.2
|
||||
final b = box(0.25, 0.25, 0.55, 0.55, 0.6); // 大框 0.3x0.3 套住小框
|
||||
final result = nms([a, b], 0.45);
|
||||
expect(result.length, 1);
|
||||
expect(result[0].score, 0.8);
|
||||
});
|
||||
|
||||
test('partiallyOverlapping_sameTarget_deduped', () {
|
||||
// 标准 IoU = 0.075/0.195 ≈ 0.385 < 0.45 会漏;minIoU = 0.075/0.09 ≈ 0.83
|
||||
final a = box(0.1, 0.1, 0.4, 0.4, 0.9); // 0.3x0.3
|
||||
final b = box(0.15, 0.1, 0.55, 0.4, 0.5); // 0.4x0.3,与 a 重叠 0.25x0.3
|
||||
final result = nms([a, b], 0.45);
|
||||
expect(result.length, 1);
|
||||
expect(result[0].score, 0.9);
|
||||
});
|
||||
|
||||
test('iou_nonOverlapping_isZero', () {
|
||||
final a = box(0.0, 0.0, 0.2, 0.2, 1);
|
||||
final b = box(0.8, 0.8, 1.0, 1.0, 1);
|
||||
|
||||
Reference in New Issue
Block a user