Commit 0c67c53
committed
fix(object-detection): don't export models with NMS and pin opset
- don't export models with NMS:
- so we can tweak NMS settings on Robotoff side if needed
- exported models didn't work out of the box (0 detections) when
tested
- pin opset to 20 for ONNX export1 parent 8ca9558 commit 0c67c53
1 file changed
+3
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
344 | 344 | | |
345 | 345 | | |
346 | 346 | | |
347 | | - | |
348 | 347 | | |
349 | 348 | | |
350 | 349 | | |
| 350 | + | |
| 351 | + | |
| 352 | + | |
351 | 353 | | |
352 | 354 | | |
353 | 355 | | |
| |||
0 commit comments