--- license: mit language: - en pipeline_tag: image-to-image tags: - Low-light-Enhancement - ImageEnhancement --- # LowLightImageEnhancement This is a collection of Low-light Enhancement algorithms, models have been converted to run on the Axera NPU using **w8a8** quantization. This model has been optimized with the following LoRA: Compatible with Pulsar2 version: 6.0 115775d3 ## Convert tools links: For those who are interested in model conversion, you can try to export axmodel through - [Pulsar2 Link, How to Convert ONNX to axmodel](https://pulsar2-docs.readthedocs.io/en/latest/pulsar2/introduction.html) ## Support Platform - AX650 - [M4N-Dock(爱芯派Pro)](https://wiki.sipeed.com/hardware/zh/maixIV/m4ndock/m4ndock.html) - [M.2 Accelerator card](https://docs.m5stack.com/en/ai_hardware/LLM-8850_Card) ## 性能基准测试 (Performance Benchmark) | Model | Input Size | Inference Time | |-------|-----------|----------------| | Zero-DCE | 256×256 | 2.6ms | | Zero-DCE++ | 512×512 | 12.7ms | | SCI | 600×400 | 1.2ms | | RetinexFormer | 224×224 | 35ms | ## How to use Download all files from this repository to the device ``` 模型文件组织方式如下: . |-- model_convert | |-- SCI.json | `-- axmodel | `-- SCI_TPAMI_600_400.axmodel |-- pic | |-- 00001.png | |-- 00051.png | |-- 00079.png | |-- 00091.png | |-- 2062.jpg | |-- 2064.jpg | |-- 3008.jpg | |-- 3018.jpg | |-- 3020.jpg | `-- NPE_71.png |-- python | |-- axmodel_infer.py | `-- onnx_infer.py `-- res `-- axmodel_res.png ``` ### Inference 图片推理,执行命令`python3 axmodel_infer.py`: ``` (base) root@ax650:~/SCI# python3 axmodel_infer.py [INFO] Available providers: ['AxEngineExecutionProvider', 'AXCLRTExecutionProvider'] [INFO] Using provider: AxEngineExecutionProvider [INFO] Chip type: ChipType.MC50 [INFO] VNPU type: VNPUType.DISABLED [INFO] Engine version: 2.12.0s [INFO] Model type: 2 (triple core) [INFO] Compiler version: 6.0-dirty 115775d3-dirty Saved comparison → ./axmodel_res.png Input: (600, 400) → model input: (600,400) → restored: (600, 400) ``` 推理结果样例: ![alt text](SCI/res/axmodel_res.png)