diff --git a/boards/OPENMV_AE3/romfs.json b/boards/OPENMV_AE3/romfs.json index 4667229b3..58c643091 100644 --- a/boards/OPENMV_AE3/romfs.json +++ b/boards/OPENMV_AE3/romfs.json @@ -22,13 +22,7 @@ }, { "type": "tflite", - "path": "{TOP}/lib/models/yolo_v5_224_nano.tflite", - "alignment": 16, - "optimize": "Performance" - }, - { - "type": "tflite", - "path": "{TOP}/lib/models/person_detect.tflite", + "path": "{TOP}/lib/models/yolo_v2_224_small.tflite", "alignment": 16, "optimize": "Performance" }, diff --git a/boot/include/port.h b/boot/include/port.h index 89c6c26dd..6c6643e7e 100644 --- a/boot/include/port.h +++ b/boot/include/port.h @@ -68,6 +68,7 @@ typedef struct { uintptr_t start; uintptr_t limit; uint32_t attr; + uint32_t delay; } partition_t; typedef enum { diff --git a/boot/src/common/dfu.c b/boot/src/common/dfu.c index a268ac6ca..41e55fff4 100644 --- a/boot/src/common/dfu.c +++ b/boot/src/common/dfu.c @@ -46,13 +46,10 @@ void tud_dfu_detach_cb(void) { // or before tud_dfu_manifest_cb() (state=DFU_MANIFEST). uint32_t tud_dfu_get_timeout_cb(uint8_t itf, uint8_t state) { tud_dfu_detached = false; + const partition_t *p = &OMV_BOOT_DFU_PARTITIONS[itf]; if (state == DFU_DNBUSY) { - if (itf < 3) { - return 0; - } else { - return 100; - } + return p->delay; } else if (state == DFU_MANIFEST) { return 0; }