tree 9e9e19df5de3563a9e71fc3f1f21e6b8675b4234
parent 69c7a43f172a969cf189bbd97b1f19956230d994
author Lun Dong <lundong@google.com> 1646161893 +0000
committer Lun Dong <lundong@google.com> 1646325164 +0000

Integrate RISP4ML into Shodan ML Toolchain using Fssd model

This CL is Part 2 of the comprehensive 3-part changes of integrating RISP4ML into Shodan ML toolchain. This change focuses on integrating RISP4ML using the FSSD model as an example.

I have verifed that the output@RISP4ML running on Qemu/Renode completely matches with the output@RISPML running on google3.

Details of the 3-part changes:

(Part 1) Completely re-wrote RISP4ML in plain C.
- https://spacebeaker-review.googlesource.com/c/shodan/sw/vec-iree/+/17180
- The original RISP4ML on google3 was written in C++. The memory usage with C++ RISP4ML was significantly larger. Thus, I have completely re-written RISP4ML in plain C. The memory increase with the plain-C RISP4ML is much smaller.
- Made the input/output shape to be configurable arguments instead of constants.
- Made low-level optimizations and simplifications.

(Part 2) Integrated plain-C RISP4ML into Shodan ML toolchain (this CL)
- Use fssd_25_8bit_v2 as an example to load the binary (raw bayer) file and go through RISP4ML toolchain, and feed the RISP4ML output into IREE flow.
- The fssd_25_8bit_v2 example is up and running, and the correctness of output has been verified (compare output@plain-C risp4ml against output@google3).

(Part 3) Added unit tests for plain-C RISP4ML
- Added unit tests based on pw_unit_test
- Build unit tests in iree_cc_binary
- Run and check unit tests via lit_test

Change-Id: I90eb31cb4b5cecfee79ad0952025ad97e12b9b52