1
This commit is contained in:
@@ -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);
|
||||
});
|
||||
|
||||
|
||||
Reference in New Issue
Block a user