mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
Compare commits
8 Commits
02c30fb3f8
...
a8c369d6f0
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a8c369d6f0 | ||
|
|
5f766069ad | ||
|
|
21ceec4229 | ||
|
|
29e94ad8e8 | ||
|
|
4b1837f72e | ||
|
|
200c702d1d | ||
|
|
9d0e012b1c | ||
|
|
b214c54c07 |
29
.github/workflows/firmware.yml
vendored
29
.github/workflows/firmware.yml
vendored
@ -48,11 +48,26 @@ jobs:
|
||||
- ARDUINO_NICLA_VISION
|
||||
- ARDUINO_NANO_RP2040_CONNECT
|
||||
- ARDUINO_NANO_33_BLE_SENSE
|
||||
- DOCKER
|
||||
profile: [0] # default profile
|
||||
qemu: [false]
|
||||
profile: [0]
|
||||
artifacts: [true]
|
||||
include:
|
||||
- target: DOCKER
|
||||
qemu: false
|
||||
profile: 0
|
||||
artifacts: false
|
||||
- target: OPENMV_N6
|
||||
qemu: false
|
||||
profile: 1
|
||||
artifacts: false
|
||||
- target: MPS2_AN500
|
||||
qemu: true
|
||||
profile: 0
|
||||
artifacts: false
|
||||
- target: MPS3_AN547
|
||||
qemu: true
|
||||
profile: 0
|
||||
artifacts: false
|
||||
fail-fast: false
|
||||
steps:
|
||||
- name: '⏳ Checkout repository'
|
||||
@ -82,6 +97,8 @@ jobs:
|
||||
|
||||
- name: '🛠 Install dependencies'
|
||||
run: |
|
||||
sudo apt-get update
|
||||
sudo apt-get install -y qemu-system-arm
|
||||
pip install -r .github/workflows/requirements.txt
|
||||
flake8 --version
|
||||
pytest --version
|
||||
@ -104,10 +121,14 @@ jobs:
|
||||
run: source tools/ci.sh && ci_install_stedgeai ${HOME}/cache/stedgeai
|
||||
|
||||
- name: '🏗 Build firmware'
|
||||
run: source tools/ci.sh && ci_build_target ${{ matrix.target }} ${{ matrix.profile }}
|
||||
run: source tools/ci.sh && ci_build_target ${{ matrix.target }} ${{ matrix.qemu }} ${{ matrix.profile }} ${{ matrix.artifacts }}
|
||||
|
||||
- name: '🧪 Run QEMU tests'
|
||||
if: matrix.qemu
|
||||
run: source tools/ci.sh && ci_run_qemu_tests ${{ matrix.target }}
|
||||
|
||||
- name: '⬆ Upload artifacts'
|
||||
if: matrix.target != 'DOCKER' && matrix.profile == 0
|
||||
if: matrix.artifacts
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: ${{ matrix.target }}
|
||||
|
||||
1
.github/workflows/requirements.txt
vendored
1
.github/workflows/requirements.txt
vendored
@ -5,3 +5,4 @@ tabulate==0.9.0
|
||||
cryptography==44.0.1
|
||||
pyelftools==0.27
|
||||
colorama==0.4.6
|
||||
mpremote
|
||||
|
||||
164
boards/MPS2_AN500/imlib_config.h
Normal file
164
boards/MPS2_AN500/imlib_config.h
Normal file
@ -0,0 +1,164 @@
|
||||
/*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*
|
||||
* Copyright (C) 2025 OpenMV, LLC.
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to deal
|
||||
* in the Software without restriction, including without limitation the rights
|
||||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in
|
||||
* all copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
* THE SOFTWARE.
|
||||
*
|
||||
* Image library configuration.
|
||||
*/
|
||||
#ifndef __IMLIB_CONFIG_H__
|
||||
#define __IMLIB_CONFIG_H__
|
||||
|
||||
// Enable Image I/O
|
||||
#define IMLIB_ENABLE_IMAGE_IO
|
||||
|
||||
// Enable Image File I/O
|
||||
#define IMLIB_ENABLE_IMAGE_FILE_IO
|
||||
|
||||
// Enable LAB LUT
|
||||
#define IMLIB_ENABLE_LAB_LUT
|
||||
|
||||
// Enable YUV LUT
|
||||
//#define IMLIB_ENABLE_YUV_LUT
|
||||
|
||||
// Enable ISP ops
|
||||
#define IMLIB_ENABLE_ISP_OPS
|
||||
|
||||
// Enable binary ops
|
||||
#define IMLIB_ENABLE_BINARY_OPS
|
||||
|
||||
// Enable math ops
|
||||
#define IMLIB_ENABLE_MATH_OPS
|
||||
|
||||
// Enable flood_fill()
|
||||
#define IMLIB_ENABLE_FLOOD_FILL
|
||||
|
||||
// Enable mean()
|
||||
#define IMLIB_ENABLE_MEAN
|
||||
|
||||
// Enable median()
|
||||
#define IMLIB_ENABLE_MEDIAN
|
||||
|
||||
// Enable mode()
|
||||
#define IMLIB_ENABLE_MODE
|
||||
|
||||
// Enable midpoint()
|
||||
#define IMLIB_ENABLE_MIDPOINT
|
||||
|
||||
// Enable morph()
|
||||
#define IMLIB_ENABLE_MORPH
|
||||
|
||||
// Enable Gaussian
|
||||
#define IMLIB_ENABLE_GAUSSIAN
|
||||
|
||||
// Enable Laplacian
|
||||
#define IMLIB_ENABLE_LAPLACIAN
|
||||
|
||||
// Enable bilateral()
|
||||
#define IMLIB_ENABLE_BILATERAL
|
||||
|
||||
// Enable linpolar()
|
||||
#define IMLIB_ENABLE_LINPOLAR
|
||||
|
||||
// Enable logpolar()
|
||||
#define IMLIB_ENABLE_LOGPOLAR
|
||||
|
||||
// Enable lens_corr()
|
||||
#define IMLIB_ENABLE_LENS_CORR
|
||||
|
||||
// Enable rotation_corr()
|
||||
#define IMLIB_ENABLE_ROTATION_CORR
|
||||
|
||||
// Enable phasecorrelate()
|
||||
#if defined(IMLIB_ENABLE_ROTATION_CORR)
|
||||
#define IMLIB_ENABLE_FIND_DISPLACEMENT
|
||||
#endif
|
||||
|
||||
// Enable get_similarity()
|
||||
#define IMLIB_ENABLE_GET_SIMILARITY
|
||||
|
||||
// Enable find_lines()
|
||||
#define IMLIB_ENABLE_FIND_LINES
|
||||
|
||||
// Enable find_line_segments()
|
||||
#define IMLIB_ENABLE_FIND_LINE_SEGMENTS
|
||||
|
||||
// Enable find_circles()
|
||||
#define IMLIB_ENABLE_FIND_CIRCLES
|
||||
|
||||
// Enable find_rects()
|
||||
#define IMLIB_ENABLE_FIND_RECTS
|
||||
|
||||
// Enable find_qrcodes() (14 KB)
|
||||
#define IMLIB_ENABLE_QRCODES
|
||||
|
||||
// Enable find_apriltags() (64 KB)
|
||||
#define IMLIB_ENABLE_APRILTAGS
|
||||
#define IMLIB_ENABLE_APRILTAGS_TAG36H11
|
||||
|
||||
// Enable fine find_apriltags() - (8-way connectivity versus 4-way connectivity)
|
||||
// #define IMLIB_ENABLE_FINE_APRILTAGS
|
||||
|
||||
// Enable high res find_apriltags() - uses more RAM
|
||||
// #define IMLIB_ENABLE_HIGH_RES_APRILTAGS
|
||||
|
||||
// Enable find_datamatrices() (26 KB)
|
||||
#define IMLIB_ENABLE_DATAMATRICES
|
||||
|
||||
// Enable find_barcodes() (42 KB)
|
||||
#define IMLIB_ENABLE_BARCODES
|
||||
|
||||
// Enable find_features() and Haar cascades.
|
||||
#define IMLIB_ENABLE_FEATURES
|
||||
|
||||
// Enable Tensor Flow
|
||||
#define IMLIB_ENABLE_STAI
|
||||
|
||||
// Enable AGAST.
|
||||
#define IMLIB_ENABLE_AGAST
|
||||
|
||||
// Enable find_template()
|
||||
#define IMLIB_FIND_TEMPLATE
|
||||
|
||||
// Enable find_lbp()
|
||||
#define IMLIB_ENABLE_FIND_LBP
|
||||
|
||||
// Enable find_keypoints()
|
||||
#if defined(IMLIB_ENABLE_FAST) || defined(IMLIB_ENABLE_AGAST)
|
||||
#define IMLIB_ENABLE_FIND_KEYPOINTS
|
||||
#endif
|
||||
|
||||
// Enable load, save and match descriptor
|
||||
#define IMLIB_ENABLE_DESCRIPTOR
|
||||
|
||||
// Enable find_hog()
|
||||
#define IMLIB_ENABLE_HOG
|
||||
|
||||
// Enable selective_search()
|
||||
#define IMLIB_ENABLE_SELECTIVE_SEARCH
|
||||
|
||||
// Enable PNG encoder/decoder
|
||||
#define IMLIB_ENABLE_PNG_ENCODER
|
||||
#define IMLIB_ENABLE_PNG_DECODER
|
||||
|
||||
// Stereo Imaging
|
||||
#define IMLIB_ENABLE_STEREO_DISPARITY
|
||||
|
||||
#endif //__IMLIB_CONFIG_H__
|
||||
5
boards/MPS2_AN500/manifest.py
Normal file
5
boards/MPS2_AN500/manifest.py
Normal file
@ -0,0 +1,5 @@
|
||||
# OpenMV library
|
||||
add_library("openmv-lib", "$(OMV_LIB_DIR)")
|
||||
|
||||
# Libraries
|
||||
require("ml", library="openmv-lib")
|
||||
71
boards/MPS2_AN500/omv_boardconfig.h
Normal file
71
boards/MPS2_AN500/omv_boardconfig.h
Normal file
@ -0,0 +1,71 @@
|
||||
/*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*
|
||||
* Copyright (C) 2025 OpenMV, LLC.
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to deal
|
||||
* in the Software without restriction, including without limitation the rights
|
||||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in
|
||||
* all copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
* THE SOFTWARE.
|
||||
*
|
||||
* Board configuration and pin definitions.
|
||||
*/
|
||||
#ifndef __OMV_BOARDCONFIG_H__
|
||||
#define __OMV_BOARDCONFIG_H__
|
||||
|
||||
// Architecture info
|
||||
#define OMV_BOARD_ARCH "MPS2 AN500" // 33 chars max
|
||||
#define OMV_BOARD_TYPE "M7"
|
||||
|
||||
// JPEG configuration.
|
||||
#define OMV_JPEG_CODEC_ENABLE (0)
|
||||
#define OMV_JPEG_QUALITY_LOW (35)
|
||||
#define OMV_JPEG_QUALITY_HIGH (60)
|
||||
#define OMV_JPEG_QUALITY_THRESHOLD (160 * 120 * 2)
|
||||
|
||||
// UMM heap block size
|
||||
#define OMV_UMM_BLOCK_SIZE 16
|
||||
|
||||
#define OMV_MAIN_MEMORY SRAM1 // Data/BSS memory
|
||||
#define OMV_STACK_MEMORY SRAM1 // stack memory
|
||||
#define OMV_STACK_SIZE (64K)
|
||||
#define OMV_FB_MEMORY SRAM1 // Framebuffer, fb_alloc
|
||||
#define OMV_FB_SIZE (1M) // FB memory: header + QVGA/GS image
|
||||
#define OMV_FB_ALLOC_SIZE (1850K) // minimum fb alloc size
|
||||
#define OMV_GC_BLOCK0_MEMORY SRAM1 // Main GC block
|
||||
#define OMV_GC_BLOCK0_SIZE (1M)
|
||||
#define OMV_SB_SIZE (128K) // IDE JPEG buffer size (header + data).
|
||||
|
||||
// Memory map.
|
||||
#define OMV_SRAM1_ORIGIN 0x20000000
|
||||
#define OMV_SRAM1_LENGTH 4M
|
||||
|
||||
#define OMV_FLASH_ISR_ORIGIN 0x00000000
|
||||
#define OMV_FLASH_ISR_LENGTH 4K
|
||||
|
||||
#define OMV_FLASH_TXT_ORIGIN 0x00001000
|
||||
#define OMV_FLASH_TXT_LENGTH 4092M
|
||||
|
||||
// CSI configuration.
|
||||
#define OMV_CSI_I2C_ID (0)
|
||||
#define OMV_CSI_I2C_SPEED (OMV_I2C_SPEED_STANDARD)
|
||||
#define OMV_CSI_CLK_SOURCE (OMV_CSI_CLK_SOURCE_TIM)
|
||||
#define OMV_CSI_CLK_FREQUENCY (24000000)
|
||||
#define OMV_CSI_HW_CROP_ENABLE (0)
|
||||
#define OMV_CSI_MAX_DEVICES (3)
|
||||
|
||||
#define OMV_SOFTCSI_ENABLE (1)
|
||||
#endif //__OMV_BOARDCONFIG_H__
|
||||
24
boards/MPS2_AN500/omv_boardconfig.mk
Executable file
24
boards/MPS2_AN500/omv_boardconfig.mk
Executable file
@ -0,0 +1,24 @@
|
||||
CPU=cortex-m7
|
||||
FPU=fpv5-d16
|
||||
PORT=qemu
|
||||
QEMU_MACHINE=mps2-an500
|
||||
OMV_ENABLE_BL=0
|
||||
OMV_LEPTON_SDK_ENABLE=0
|
||||
OMV_BOARD_CFLAGS= -DQEMU_SOC_MPS2 \
|
||||
-DCPU_FREQ_HZ=25000000
|
||||
MICROPY_PY_CSI = 1
|
||||
MICROPY_PY_CSI_NG = 1
|
||||
MICROPY_PY_FIR = 0
|
||||
MICROPY_PY_ULAB = 1
|
||||
MICROPY_PY_WINC1500 = 0
|
||||
MICROPY_PY_DISPLAY = 0
|
||||
MICROPY_PY_TV = 0
|
||||
MICROPY_PY_ML = 1
|
||||
MICROPY_PY_ML_TFLM = 1
|
||||
MICROPY_PY_LWIP = 0
|
||||
MICROPY_PY_SSL = 0
|
||||
MICROPY_PY_SSL_ECDSA_SIGN_ALT = 0
|
||||
MICROPY_SSL_MBEDTLS = 0
|
||||
MICROPY_PY_NETWORK_CYW43 = 0
|
||||
MICROPY_PY_BLUETOOTH = 0
|
||||
MICROPY_BLUETOOTH_NIMBLE = 0
|
||||
6
boards/MPS2_AN500/romfs.json
Normal file
6
boards/MPS2_AN500/romfs.json
Normal file
@ -0,0 +1,6 @@
|
||||
{
|
||||
"0": {
|
||||
"size": "0x20000",
|
||||
"entries": [ ]
|
||||
}
|
||||
}
|
||||
17
boards/MPS2_AN500/ulab_config.h
Normal file
17
boards/MPS2_AN500/ulab_config.h
Normal file
@ -0,0 +1,17 @@
|
||||
/*
|
||||
* This file is part of the OpenMV project.
|
||||
* Copyright (c) 2013-2016 Kwabena W. Agyeman <kwagyeman@openmv.io>
|
||||
* This work is licensed under the MIT license, see the file LICENSE for details.
|
||||
*
|
||||
* Ulab config file.
|
||||
*
|
||||
*/
|
||||
#ifndef __ULAB_CONFIG_H__
|
||||
#define __ULAB_CONFIG_H__
|
||||
// Override ulab defaults here.
|
||||
#define ULAB_MAX_DIMS (4)
|
||||
#define ULAB_SUPPORTS_COMPLEX (0)
|
||||
#define ULAB_SCIPY_HAS_OPTIMIZE_MODULE (0)
|
||||
#define ULAB_SCIPY_HAS_SPECIAL_MODULE (0)
|
||||
#define ULAB_FFT_IS_NUMPY_COMPATIBLE (0)
|
||||
#endif //__ULAB_CONFIG_H__
|
||||
164
boards/MPS3_AN547/imlib_config.h
Normal file
164
boards/MPS3_AN547/imlib_config.h
Normal file
@ -0,0 +1,164 @@
|
||||
/*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*
|
||||
* Copyright (C) 2025 OpenMV, LLC.
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to deal
|
||||
* in the Software without restriction, including without limitation the rights
|
||||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in
|
||||
* all copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
* THE SOFTWARE.
|
||||
*
|
||||
* Image library configuration.
|
||||
*/
|
||||
#ifndef __IMLIB_CONFIG_H__
|
||||
#define __IMLIB_CONFIG_H__
|
||||
|
||||
// Enable Image I/O
|
||||
#define IMLIB_ENABLE_IMAGE_IO
|
||||
|
||||
// Enable Image File I/O
|
||||
#define IMLIB_ENABLE_IMAGE_FILE_IO
|
||||
|
||||
// Enable LAB LUT
|
||||
#define IMLIB_ENABLE_LAB_LUT
|
||||
|
||||
// Enable YUV LUT
|
||||
//#define IMLIB_ENABLE_YUV_LUT
|
||||
|
||||
// Enable ISP ops
|
||||
#define IMLIB_ENABLE_ISP_OPS
|
||||
|
||||
// Enable binary ops
|
||||
#define IMLIB_ENABLE_BINARY_OPS
|
||||
|
||||
// Enable math ops
|
||||
#define IMLIB_ENABLE_MATH_OPS
|
||||
|
||||
// Enable flood_fill()
|
||||
#define IMLIB_ENABLE_FLOOD_FILL
|
||||
|
||||
// Enable mean()
|
||||
#define IMLIB_ENABLE_MEAN
|
||||
|
||||
// Enable median()
|
||||
#define IMLIB_ENABLE_MEDIAN
|
||||
|
||||
// Enable mode()
|
||||
#define IMLIB_ENABLE_MODE
|
||||
|
||||
// Enable midpoint()
|
||||
#define IMLIB_ENABLE_MIDPOINT
|
||||
|
||||
// Enable morph()
|
||||
#define IMLIB_ENABLE_MORPH
|
||||
|
||||
// Enable Gaussian
|
||||
#define IMLIB_ENABLE_GAUSSIAN
|
||||
|
||||
// Enable Laplacian
|
||||
#define IMLIB_ENABLE_LAPLACIAN
|
||||
|
||||
// Enable bilateral()
|
||||
#define IMLIB_ENABLE_BILATERAL
|
||||
|
||||
// Enable linpolar()
|
||||
#define IMLIB_ENABLE_LINPOLAR
|
||||
|
||||
// Enable logpolar()
|
||||
#define IMLIB_ENABLE_LOGPOLAR
|
||||
|
||||
// Enable lens_corr()
|
||||
#define IMLIB_ENABLE_LENS_CORR
|
||||
|
||||
// Enable rotation_corr()
|
||||
#define IMLIB_ENABLE_ROTATION_CORR
|
||||
|
||||
// Enable phasecorrelate()
|
||||
#if defined(IMLIB_ENABLE_ROTATION_CORR)
|
||||
#define IMLIB_ENABLE_FIND_DISPLACEMENT
|
||||
#endif
|
||||
|
||||
// Enable get_similarity()
|
||||
#define IMLIB_ENABLE_GET_SIMILARITY
|
||||
|
||||
// Enable find_lines()
|
||||
#define IMLIB_ENABLE_FIND_LINES
|
||||
|
||||
// Enable find_line_segments()
|
||||
#define IMLIB_ENABLE_FIND_LINE_SEGMENTS
|
||||
|
||||
// Enable find_circles()
|
||||
#define IMLIB_ENABLE_FIND_CIRCLES
|
||||
|
||||
// Enable find_rects()
|
||||
#define IMLIB_ENABLE_FIND_RECTS
|
||||
|
||||
// Enable find_qrcodes() (14 KB)
|
||||
#define IMLIB_ENABLE_QRCODES
|
||||
|
||||
// Enable find_apriltags() (64 KB)
|
||||
#define IMLIB_ENABLE_APRILTAGS
|
||||
#define IMLIB_ENABLE_APRILTAGS_TAG36H11
|
||||
|
||||
// Enable fine find_apriltags() - (8-way connectivity versus 4-way connectivity)
|
||||
// #define IMLIB_ENABLE_FINE_APRILTAGS
|
||||
|
||||
// Enable high res find_apriltags() - uses more RAM
|
||||
// #define IMLIB_ENABLE_HIGH_RES_APRILTAGS
|
||||
|
||||
// Enable find_datamatrices() (26 KB)
|
||||
#define IMLIB_ENABLE_DATAMATRICES
|
||||
|
||||
// Enable find_barcodes() (42 KB)
|
||||
#define IMLIB_ENABLE_BARCODES
|
||||
|
||||
// Enable find_features() and Haar cascades.
|
||||
#define IMLIB_ENABLE_FEATURES
|
||||
|
||||
// Enable Tensor Flow
|
||||
#define IMLIB_ENABLE_STAI
|
||||
|
||||
// Enable AGAST.
|
||||
#define IMLIB_ENABLE_AGAST
|
||||
|
||||
// Enable find_template()
|
||||
#define IMLIB_FIND_TEMPLATE
|
||||
|
||||
// Enable find_lbp()
|
||||
#define IMLIB_ENABLE_FIND_LBP
|
||||
|
||||
// Enable find_keypoints()
|
||||
#if defined(IMLIB_ENABLE_FAST) || defined(IMLIB_ENABLE_AGAST)
|
||||
#define IMLIB_ENABLE_FIND_KEYPOINTS
|
||||
#endif
|
||||
|
||||
// Enable load, save and match descriptor
|
||||
#define IMLIB_ENABLE_DESCRIPTOR
|
||||
|
||||
// Enable find_hog()
|
||||
#define IMLIB_ENABLE_HOG
|
||||
|
||||
// Enable selective_search()
|
||||
#define IMLIB_ENABLE_SELECTIVE_SEARCH
|
||||
|
||||
// Enable PNG encoder/decoder
|
||||
#define IMLIB_ENABLE_PNG_ENCODER
|
||||
#define IMLIB_ENABLE_PNG_DECODER
|
||||
|
||||
// Stereo Imaging
|
||||
#define IMLIB_ENABLE_STEREO_DISPARITY
|
||||
|
||||
#endif //__IMLIB_CONFIG_H__
|
||||
5
boards/MPS3_AN547/manifest.py
Normal file
5
boards/MPS3_AN547/manifest.py
Normal file
@ -0,0 +1,5 @@
|
||||
# OpenMV library
|
||||
add_library("openmv-lib", "$(OMV_LIB_DIR)")
|
||||
|
||||
# Libraries
|
||||
require("ml", library="openmv-lib")
|
||||
75
boards/MPS3_AN547/omv_boardconfig.h
Normal file
75
boards/MPS3_AN547/omv_boardconfig.h
Normal file
@ -0,0 +1,75 @@
|
||||
/*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*
|
||||
* Copyright (C) 2025 OpenMV, LLC.
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to deal
|
||||
* in the Software without restriction, including without limitation the rights
|
||||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in
|
||||
* all copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
* THE SOFTWARE.
|
||||
*
|
||||
* Board configuration and pin definitions.
|
||||
*/
|
||||
#ifndef __OMV_BOARDCONFIG_H__
|
||||
#define __OMV_BOARDCONFIG_H__
|
||||
|
||||
// Architecture info
|
||||
#define OMV_BOARD_ARCH "MPS3 AN547" // 33 chars max
|
||||
#define OMV_BOARD_TYPE "M55"
|
||||
|
||||
// JPEG configuration.
|
||||
#define OMV_JPEG_CODEC_ENABLE (0)
|
||||
#define OMV_JPEG_QUALITY_LOW (35)
|
||||
#define OMV_JPEG_QUALITY_HIGH (60)
|
||||
#define OMV_JPEG_QUALITY_THRESHOLD (160 * 120 * 2)
|
||||
|
||||
// UMM heap block size
|
||||
#define OMV_UMM_BLOCK_SIZE 16
|
||||
|
||||
#define OMV_MAIN_MEMORY SRAM1 // Data/BSS memory
|
||||
#define OMV_STACK_MEMORY SRAM1 // stack memory
|
||||
#define OMV_STACK_SIZE (64K)
|
||||
#define OMV_FB_MEMORY SRAM1 // Framebuffer, fb_alloc
|
||||
#define OMV_FB_SIZE (1M) // FB memory: header + QVGA/GS image
|
||||
#define OMV_FB_ALLOC_SIZE (1850K) // minimum fb alloc size
|
||||
#define OMV_GC_BLOCK0_MEMORY SRAM1 // Main GC block
|
||||
#define OMV_GC_BLOCK0_SIZE (1M)
|
||||
#define OMV_SB_SIZE (128K) // IDE JPEG buffer size (header + data).
|
||||
|
||||
// Memory map.
|
||||
#define OMV_ITCM_ORIGIN 0x00000000
|
||||
#define OMV_ITCM_LENGTH 512K
|
||||
|
||||
#define OMV_SRAM1_ORIGIN 0x21000000
|
||||
#define OMV_SRAM1_LENGTH 4M
|
||||
|
||||
#define OMV_FLASH_ISR_ORIGIN 0x00000000
|
||||
#define OMV_FLASH_ISR_LENGTH 4K
|
||||
|
||||
#define OMV_FLASH_TXT_ORIGIN 0x01000000
|
||||
#define OMV_FLASH_TXT_LENGTH 2M
|
||||
|
||||
// CSI configuration.
|
||||
#define OMV_CSI_I2C_ID (0)
|
||||
#define OMV_CSI_I2C_SPEED (OMV_I2C_SPEED_STANDARD)
|
||||
#define OMV_CSI_CLK_SOURCE (OMV_CSI_CLK_SOURCE_TIM)
|
||||
#define OMV_CSI_CLK_FREQUENCY (24000000)
|
||||
#define OMV_CSI_HW_CROP_ENABLE (0)
|
||||
#define OMV_CSI_MAX_DEVICES (3)
|
||||
|
||||
#define OMV_SOFTCSI_ENABLE (1)
|
||||
|
||||
#endif //__OMV_BOARDCONFIG_H__
|
||||
25
boards/MPS3_AN547/omv_boardconfig.mk
Normal file
25
boards/MPS3_AN547/omv_boardconfig.mk
Normal file
@ -0,0 +1,25 @@
|
||||
CPU=cortex-m55
|
||||
FPU=auto
|
||||
PORT=qemu
|
||||
QEMU_MACHINE=mps3-an547
|
||||
OMV_ENABLE_BL=0
|
||||
OMV_LEPTON_SDK_ENABLE=0
|
||||
OMV_BOARD_CFLAGS= -mcmse \
|
||||
-DQEMU_SOC_MPS3 \
|
||||
-DCPU_FREQ_HZ=32000000
|
||||
MICROPY_PY_CSI = 1
|
||||
MICROPY_PY_CSI_NG = 1
|
||||
MICROPY_PY_FIR = 0
|
||||
MICROPY_PY_ULAB = 1
|
||||
MICROPY_PY_WINC1500 = 0
|
||||
MICROPY_PY_DISPLAY = 0
|
||||
MICROPY_PY_TV = 0
|
||||
MICROPY_PY_ML = 1
|
||||
MICROPY_PY_ML_TFLM = 1
|
||||
MICROPY_PY_LWIP = 0
|
||||
MICROPY_PY_SSL = 0
|
||||
MICROPY_PY_SSL_ECDSA_SIGN_ALT = 0
|
||||
MICROPY_SSL_MBEDTLS = 0
|
||||
MICROPY_PY_NETWORK_CYW43 = 0
|
||||
MICROPY_PY_BLUETOOTH = 0
|
||||
MICROPY_BLUETOOTH_NIMBLE = 0
|
||||
6
boards/MPS3_AN547/romfs.json
Normal file
6
boards/MPS3_AN547/romfs.json
Normal file
@ -0,0 +1,6 @@
|
||||
{
|
||||
"0": {
|
||||
"size": "0x20000",
|
||||
"entries": [ ]
|
||||
}
|
||||
}
|
||||
17
boards/MPS3_AN547/ulab_config.h
Normal file
17
boards/MPS3_AN547/ulab_config.h
Normal file
@ -0,0 +1,17 @@
|
||||
/*
|
||||
* This file is part of the OpenMV project.
|
||||
* Copyright (c) 2013-2016 Kwabena W. Agyeman <kwagyeman@openmv.io>
|
||||
* This work is licensed under the MIT license, see the file LICENSE for details.
|
||||
*
|
||||
* Ulab config file.
|
||||
*
|
||||
*/
|
||||
#ifndef __ULAB_CONFIG_H__
|
||||
#define __ULAB_CONFIG_H__
|
||||
// Override ulab defaults here.
|
||||
#define ULAB_MAX_DIMS (4)
|
||||
#define ULAB_SUPPORTS_COMPLEX (0)
|
||||
#define ULAB_SCIPY_HAS_OPTIMIZE_MODULE (0)
|
||||
#define ULAB_SCIPY_HAS_SPECIAL_MODULE (0)
|
||||
#define ULAB_FFT_IS_NUMPY_COMPATIBLE (0)
|
||||
#endif //__ULAB_CONFIG_H__
|
||||
@ -23,7 +23,7 @@
|
||||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
* THE SOFTWARE.
|
||||
*/
|
||||
|
||||
#ifndef QEMU_BUILD
|
||||
#include <stdlib.h>
|
||||
#include "py/mphal.h"
|
||||
#include "py/runtime.h"
|
||||
@ -192,3 +192,4 @@ __attribute__((naked)) void PendSV_Handler(void) {
|
||||
"nlr_jump_ptr: .word nlr_jump\n"
|
||||
);
|
||||
}
|
||||
#endif // QEMU_BUILD
|
||||
|
||||
@ -23,6 +23,7 @@
|
||||
*
|
||||
* USB debugger.
|
||||
*/
|
||||
#ifndef QEMU_BUILD
|
||||
#include <string.h>
|
||||
#include <stdio.h>
|
||||
#include "py/nlr.h"
|
||||
@ -452,3 +453,4 @@ void usbdbg_control(void *buffer, uint8_t request, uint32_t size) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
#endif // QEMU_BUILD
|
||||
|
||||
@ -29,10 +29,8 @@ ENV MAKE_URL="https://ftp.gnu.org/gnu/make/make-4.4.1.tar.gz"
|
||||
RUN wget --no-check-certificate --user-agent="Mozilla/5.0" -O - ${MAKE_URL} | tar --strip-components=1 -xz -C /workspace/make
|
||||
RUN cd /workspace/make && ./configure && make -j$(nproc)
|
||||
|
||||
# Set up directories
|
||||
WORKDIR /workspace
|
||||
|
||||
COPY . .
|
||||
|
||||
# Permissive git permissions inside container
|
||||
ENV HOME=/tmp
|
||||
RUN git config --global --add safe.directory '*'
|
||||
|
||||
ENV PATH="/workspace/gcc/bin:/workspace/llvm/bin:/workspace/cmake/bin:/workspace/make:$PATH"
|
||||
|
||||
@ -3,6 +3,27 @@ DOCKER_TAG = latest
|
||||
CONTAINER_NAME = firmware-container
|
||||
DOCKERFILE_PATH = Dockerfile
|
||||
|
||||
# Detect repository structure
|
||||
REPO_ROOT := $(shell git rev-parse --show-toplevel)
|
||||
GIT_DIR := $(shell realpath $$(git rev-parse --git-dir))
|
||||
GIT_COMMON_DIR := $(shell realpath $$(git rev-parse --git-common-dir))
|
||||
|
||||
# Detect worktree: git-dir != git-common-dir
|
||||
IS_WORKTREE := $(shell [ "$(GIT_DIR)" != "$(GIT_COMMON_DIR)" ] && echo "yes" || echo "no")
|
||||
|
||||
# Working directory inside container
|
||||
WORKDIR = $(REPO_ROOT)
|
||||
|
||||
# Volume mounts: always mount repo at actual path
|
||||
ifeq ($(IS_WORKTREE),yes)
|
||||
# For worktrees, also mount the main repo (for .git references)
|
||||
MAIN_REPO_PATH := $(shell dirname $(GIT_COMMON_DIR))
|
||||
VOLUME_MOUNTS = -v $(MAIN_REPO_PATH)/.git:$(MAIN_REPO_PATH)/.git \
|
||||
-v $(REPO_ROOT):$(WORKDIR)
|
||||
else
|
||||
VOLUME_MOUNTS = -v $(REPO_ROOT):$(WORKDIR)
|
||||
endif
|
||||
|
||||
# Build the Docker image
|
||||
build-image:
|
||||
docker build \
|
||||
@ -15,7 +36,8 @@ build-firmware: build-image
|
||||
-e TARGET=$(TARGET) \
|
||||
-e HOST_UID=$(shell id -u) \
|
||||
-e HOST_GID=$(shell id -g) \
|
||||
-v $(PWD)/build:/workspace/build \
|
||||
-w $(WORKDIR) \
|
||||
$(VOLUME_MOUNTS) \
|
||||
--name $(CONTAINER_NAME) \
|
||||
$(DOCKER_IMAGE_NAME):$(DOCKER_TAG) docker/build.sh
|
||||
|
||||
@ -23,7 +45,18 @@ build-firmware: build-image
|
||||
shell:
|
||||
docker run --rm -it \
|
||||
-e shell="true" \
|
||||
-v $(PWD)/build:/workspace/build \
|
||||
-w $(WORKDIR) \
|
||||
$(VOLUME_MOUNTS) \
|
||||
$(DOCKER_IMAGE_NAME):$(DOCKER_TAG) bash
|
||||
|
||||
# Debug target to show detected paths
|
||||
debug-paths:
|
||||
@echo "IS_WORKTREE: $(IS_WORKTREE)"
|
||||
@echo "REPO_ROOT: $(REPO_ROOT)"
|
||||
@echo "GIT_DIR: $(GIT_DIR)"
|
||||
@echo "GIT_COMMON_DIR: $(GIT_COMMON_DIR)"
|
||||
@echo "MAIN_REPO_PATH: $(MAIN_REPO_PATH)"
|
||||
@echo "WORKDIR: $(WORKDIR)"
|
||||
@echo "VOLUME_MOUNTS: $(VOLUME_MOUNTS)"
|
||||
|
||||
.DEFAULT_GOAL := build-firmware
|
||||
|
||||
@ -1,7 +1,8 @@
|
||||
#!/bin/bash
|
||||
set -e -x
|
||||
|
||||
BUILD_DIR=/workspace/build/${TARGET}
|
||||
OPENMV="$(pwd)"
|
||||
BUILD_DIR=${OPENMV}/build/${TARGET}
|
||||
|
||||
# Update submodules.
|
||||
git submodule update --init --depth=1
|
||||
@ -13,4 +14,4 @@ make -j$(nproc) -C lib/micropython/mpy-cross
|
||||
make -j$(nproc) BUILD=${BUILD_DIR} TARGET=${TARGET} LLVM_PATH=/workspace/llvm/bin
|
||||
|
||||
# Fix permissions.
|
||||
chown -R ${HOST_UID:-1000}:${HOST_GID:-1000} /workspace/build
|
||||
chown -R ${HOST_UID:-1000}:${HOST_GID:-1000} ${OPENMV}/build
|
||||
|
||||
@ -35,7 +35,10 @@
|
||||
#include "omv_i2c.h"
|
||||
#include "framebuffer.h"
|
||||
|
||||
static uint32_t step = 0;
|
||||
|
||||
static int reset(omv_csi_t *csi) {
|
||||
step = 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
@ -88,7 +91,6 @@ static int snapshot(omv_csi_t *csi, image_t *image, uint32_t flags) {
|
||||
framebuffer_to_image(fb, image);
|
||||
image->pixels = buffer->data;
|
||||
|
||||
static uint32_t step = 0;
|
||||
uint32_t offset = (step++ / 4);
|
||||
|
||||
for (size_t y = 0; y < image->h; y++) {
|
||||
|
||||
@ -464,7 +464,12 @@ static mp_obj_t py_imageio_sync(mp_obj_t self) {
|
||||
static MP_DEFINE_CONST_FUN_OBJ_1(py_imageio_sync_obj, py_imageio_sync);
|
||||
|
||||
static mp_obj_t py_imageio_close(mp_obj_t self) {
|
||||
py_imageio_obj_t *stream = py_imageio_obj(self);
|
||||
py_imageio_obj_t *stream = MP_OBJ_TO_PTR(self);
|
||||
|
||||
if (stream->closed) {
|
||||
// Already closed
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
if (0) {
|
||||
#if defined(IMLIB_ENABLE_IMAGE_FILE_IO)
|
||||
@ -477,7 +482,7 @@ static mp_obj_t py_imageio_close(mp_obj_t self) {
|
||||
|
||||
stream->closed = true;
|
||||
|
||||
return self;
|
||||
return mp_const_none;
|
||||
}
|
||||
static MP_DEFINE_CONST_FUN_OBJ_1(py_imageio_close_obj, py_imageio_close);
|
||||
|
||||
|
||||
@ -55,7 +55,8 @@ static mp_obj_t py_omv_board_type() {
|
||||
static MP_DEFINE_CONST_FUN_OBJ_0(py_omv_board_type_obj, py_omv_board_type);
|
||||
|
||||
static mp_obj_t py_omv_board_id() {
|
||||
char str[25];
|
||||
char str[25] = {0};
|
||||
#ifdef OMV_BOARD_UID_ADDR
|
||||
snprintf(str, 25, "%08X%08X%08X",
|
||||
#if (OMV_BOARD_UID_SIZE == 2)
|
||||
0U,
|
||||
@ -64,6 +65,7 @@ static mp_obj_t py_omv_board_id() {
|
||||
#endif
|
||||
*((unsigned int *) (OMV_BOARD_UID_ADDR + OMV_BOARD_UID_OFFSET * 1)),
|
||||
*((unsigned int *) (OMV_BOARD_UID_ADDR + OMV_BOARD_UID_OFFSET * 0)));
|
||||
#endif
|
||||
return mp_obj_new_str(str, strlen(str));
|
||||
}
|
||||
static MP_DEFINE_CONST_FUN_OBJ_0(py_omv_board_id_obj, py_omv_board_id);
|
||||
|
||||
118
ports/qemu/main.c
Normal file
118
ports/qemu/main.c
Normal file
@ -0,0 +1,118 @@
|
||||
/*
|
||||
* Copyright (C) 2023-2024 OpenMV, LLC.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in
|
||||
* the documentation and/or other materials provided with the
|
||||
* distribution.
|
||||
* 3. Any redistribution, use, or modification in source or binary form
|
||||
* is done solely for personal benefit and not for any commercial
|
||||
* purpose or for monetary gain. For commercial licensing options,
|
||||
* please contact openmv@openmv.io
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE LICENSOR AND COPYRIGHT OWNER "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
||||
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE LICENSOR OR COPYRIGHT
|
||||
* OWNER BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
||||
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
#include "py/compile.h"
|
||||
#include "py/runtime.h"
|
||||
#include "py/gc.h"
|
||||
#include "py/mperrno.h"
|
||||
#include "py/mphal.h"
|
||||
#include "py/stackctrl.h"
|
||||
#include "shared/readline/readline.h"
|
||||
#include "shared/runtime/gchelper.h"
|
||||
#include "shared/runtime/pyexec.h"
|
||||
|
||||
#include "omv_boardconfig.h"
|
||||
#include "mp_utils.h"
|
||||
#include "file_utils.h"
|
||||
#include "fb_alloc.h"
|
||||
#include "framebuffer.h"
|
||||
#include "omv_csi.h"
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
bool first_soft_reset = true;
|
||||
|
||||
soft_reset:
|
||||
// Initialise stack extents and GC heap.
|
||||
extern uint8_t _estack, _sstack, _heap_start, _heap_end;
|
||||
mp_init_gc_stack(&_sstack, &_estack, &_heap_start, &_heap_end, 1024);
|
||||
|
||||
mp_init();
|
||||
|
||||
fb_alloc_init0();
|
||||
framebuffer_init0();
|
||||
|
||||
#ifdef IMLIB_ENABLE_IMAGE_FILE_IO
|
||||
file_buffer_init0();
|
||||
#endif
|
||||
|
||||
imlib_init_all();
|
||||
|
||||
#if MICROPY_PY_CSI
|
||||
omv_csi_init0();
|
||||
|
||||
if (first_soft_reset) {
|
||||
omv_csi_init();
|
||||
}
|
||||
#endif
|
||||
|
||||
for (;;) {
|
||||
if (pyexec_mode_kind == PYEXEC_MODE_RAW_REPL) {
|
||||
if (pyexec_raw_repl() != 0) {
|
||||
break;
|
||||
}
|
||||
} else {
|
||||
if (pyexec_friendly_repl() != 0) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
mp_printf(&mp_plat_print, "MPY: soft reboot\n");
|
||||
gc_sweep_all();
|
||||
mp_deinit();
|
||||
first_soft_reset = false;
|
||||
goto soft_reset;
|
||||
}
|
||||
|
||||
void gc_collect(void) {
|
||||
gc_collect_start();
|
||||
gc_helper_collect_regs_and_stack();
|
||||
gc_collect_end();
|
||||
}
|
||||
|
||||
void nlr_jump_fail(void *val) {
|
||||
mp_printf(&mp_plat_print, "FATAL: uncaught exception %p\n", val);
|
||||
mp_obj_print_exception(&mp_plat_print, MP_OBJ_FROM_PTR(val));
|
||||
exit(1);
|
||||
}
|
||||
|
||||
void __fatal_error(const char *msg) {
|
||||
mp_hal_stdout_tx_strn("\nFATAL ERROR:\n", 14);
|
||||
mp_hal_stdout_tx_strn(msg, strlen(msg));
|
||||
for (;;) {
|
||||
*(volatile uint32_t *) 0x4900C000 ^= 9;
|
||||
for (volatile uint delay = 0; delay < 10000000; delay++) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void usbdbg_set_irq_enabled(bool enabled) {
|
||||
|
||||
}
|
||||
85
ports/qemu/omv_csi.c
Normal file
85
ports/qemu/omv_csi.c
Normal file
@ -0,0 +1,85 @@
|
||||
/*
|
||||
* Copyright (C) 2023-2025 OpenMV, LLC.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in
|
||||
* the documentation and/or other materials provided with the
|
||||
* distribution.
|
||||
* 3. Any redistribution, use, or modification in source or binary form
|
||||
* is done solely for personal benefit and not for any commercial
|
||||
* purpose or for monetary gain. For commercial licensing options,
|
||||
* please contact openmv@openmv.io
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE LICENSOR AND COPYRIGHT OWNER "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
||||
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE LICENSOR OR COPYRIGHT
|
||||
* OWNER BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
||||
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*
|
||||
* QEMU dummy CSI driver.
|
||||
*/
|
||||
#if MICROPY_PY_CSI
|
||||
#include <string.h>
|
||||
#include <stdint.h>
|
||||
#include <stdbool.h>
|
||||
#include <stdio.h>
|
||||
|
||||
#include "py/mphal.h"
|
||||
#include "py/runtime.h"
|
||||
|
||||
#include "omv_boardconfig.h"
|
||||
#include "omv_i2c.h"
|
||||
#include "omv_csi.h"
|
||||
|
||||
static int qemu_csi_config(omv_csi_t *csi, omv_csi_config_t config) {
|
||||
// Dummy implementation - accept all configurations
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int qemu_csi_abort(omv_csi_t *csi, bool fifo_flush, bool in_irq) {
|
||||
// Dummy implementation
|
||||
return 0;
|
||||
}
|
||||
|
||||
static uint32_t qemu_clk_get_frequency(omv_clk_t *clk) {
|
||||
// Dummy implementation - return nominal frequency
|
||||
return clk->freq;
|
||||
}
|
||||
|
||||
static int qemu_clk_set_frequency(omv_clk_t *clk, uint32_t frequency) {
|
||||
// Dummy implementation - accept any frequency
|
||||
clk->freq = frequency;
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int qemu_csi_snapshot(omv_csi_t *csi, image_t *dst_image, uint32_t flags) {
|
||||
// Dummy implementation - return timeout error since there's no camera
|
||||
return OMV_CSI_ERROR_CAPTURE_TIMEOUT;
|
||||
}
|
||||
|
||||
int omv_csi_ops_init(omv_csi_t *csi) {
|
||||
// Set CSI ops to dummy implementations
|
||||
csi->abort = qemu_csi_abort;
|
||||
csi->config = qemu_csi_config;
|
||||
csi->snapshot = qemu_csi_snapshot;
|
||||
|
||||
// Set CSI clock ops
|
||||
csi->clk->freq = OMV_CSI_CLK_FREQUENCY;
|
||||
csi->clk->set_freq = qemu_clk_set_frequency;
|
||||
csi->clk->get_freq = qemu_clk_get_frequency;
|
||||
|
||||
return 0;
|
||||
}
|
||||
#endif // MICROPY_PY_CSI
|
||||
68
ports/qemu/omv_i2c.c
Normal file
68
ports/qemu/omv_i2c.c
Normal file
@ -0,0 +1,68 @@
|
||||
/*
|
||||
* Copyright (C) 2023-2025 OpenMV, LLC.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in
|
||||
* the documentation and/or other materials provided with the
|
||||
* distribution.
|
||||
* 3. Any redistribution, use, or modification in source or binary form
|
||||
* is done solely for personal benefit and not for any commercial
|
||||
* purpose or for monetary gain. For commercial licensing options,
|
||||
* please contact openmv@openmv.io
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE LICENSOR AND COPYRIGHT OWNER "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
||||
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE LICENSOR OR COPYRIGHT
|
||||
* OWNER BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
||||
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*
|
||||
* QEMU dummy I2C driver.
|
||||
*/
|
||||
#include <stdio.h>
|
||||
#include <stdbool.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "py/mphal.h"
|
||||
#include "py/runtime.h"
|
||||
|
||||
#include "omv_portconfig.h"
|
||||
#include "omv_i2c.h"
|
||||
|
||||
int omv_i2c_init(omv_i2c_t *i2c, uint32_t bus_id, uint32_t speed) {
|
||||
i2c->id = bus_id;
|
||||
i2c->speed = speed;
|
||||
i2c->initialized = true;
|
||||
i2c->scl_pin = NULL;
|
||||
i2c->sda_pin = NULL;
|
||||
i2c->inst = NULL;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int omv_i2c_deinit(omv_i2c_t *i2c) {
|
||||
if (i2c->initialized) {
|
||||
i2c->initialized = false;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int omv_i2c_scan(omv_i2c_t *i2c, uint8_t *list, uint8_t size) {
|
||||
// Dummy implementation - no devices found
|
||||
return 0;
|
||||
}
|
||||
|
||||
int omv_i2c_enable(omv_i2c_t *i2c, bool enable) {
|
||||
// Dummy implementation
|
||||
return 0;
|
||||
}
|
||||
50
ports/qemu/omv_mpconfigport.h
Normal file
50
ports/qemu/omv_mpconfigport.h
Normal file
@ -0,0 +1,50 @@
|
||||
/*
|
||||
* Copyright (C) 2023-2024 OpenMV, LLC.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in
|
||||
* the documentation and/or other materials provided with the
|
||||
* distribution.
|
||||
* 3. Any redistribution, use, or modification in source or binary form
|
||||
* is done solely for personal benefit and not for any commercial
|
||||
* purpose or for monetary gain. For commercial licensing options,
|
||||
* please contact openmv@openmv.io
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE LICENSOR AND COPYRIGHT OWNER "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
||||
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE LICENSOR OR COPYRIGHT
|
||||
* OWNER BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
||||
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*
|
||||
* MicroPython port configuration.
|
||||
*/
|
||||
#define MICROPY_NLR_RAISE_HOOK \
|
||||
do { \
|
||||
extern void fb_alloc_free_till_mark(); \
|
||||
fb_alloc_free_till_mark(); \
|
||||
} while (0);
|
||||
|
||||
#define MICROPY_OBJ_REPR (MICROPY_OBJ_REPR_C)
|
||||
typedef int mp_int_t; // must be pointer size
|
||||
typedef unsigned int mp_uint_t; // must be pointer size
|
||||
|
||||
#define MICROPY_PY_TIME (1)
|
||||
#define MICROPY_GC_SPLIT_HEAP (1)
|
||||
#define MICROPY_BANNER_NAME_AND_VERSION "OpenMV " OPENMV_GIT_TAG "; MicroPython " MICROPY_GIT_TAG
|
||||
|
||||
#include <mpconfigport.h>
|
||||
|
||||
#undef MICROPY_MEM_STATS
|
||||
#undef MICROPY_MALLOC_USES_ALLOCATED_SIZE
|
||||
59
ports/qemu/omv_portconfig.h
Normal file
59
ports/qemu/omv_portconfig.h
Normal file
@ -0,0 +1,59 @@
|
||||
/*
|
||||
* Copyright (C) 2023-2025 OpenMV, LLC.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in
|
||||
* the documentation and/or other materials provided with the
|
||||
* distribution.
|
||||
* 3. Any redistribution, use, or modification in source or binary form
|
||||
* is done solely for personal benefit and not for any commercial
|
||||
* purpose or for monetary gain. For commercial licensing options,
|
||||
* please contact openmv@openmv.io
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE LICENSOR AND COPYRIGHT OWNER "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
||||
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE LICENSOR OR COPYRIGHT
|
||||
* OWNER BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
||||
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*
|
||||
* OpenMV QEMU port abstraction layer.
|
||||
*/
|
||||
#ifndef __OMV_PORTCONFIG_H__
|
||||
#define __OMV_PORTCONFIG_H__
|
||||
#include <stdlib.h>
|
||||
|
||||
// omv_gpio_t definitions
|
||||
typedef void *omv_gpio_t;
|
||||
|
||||
// GPIO modes, speeds, and pull configurations (dummy values for QEMU)
|
||||
#define OMV_GPIO_MODE_INPUT (0)
|
||||
#define OMV_GPIO_MODE_OUTPUT (1)
|
||||
#define OMV_GPIO_MODE_OUTPUT_OD (2)
|
||||
#define OMV_GPIO_MODE_ALT (3)
|
||||
#define OMV_GPIO_MODE_ALT_OD (4)
|
||||
|
||||
#define OMV_GPIO_PULL_NONE (0)
|
||||
#define OMV_GPIO_PULL_UP (1)
|
||||
#define OMV_GPIO_PULL_DOWN (2)
|
||||
|
||||
#define OMV_GPIO_SPEED_LOW (0)
|
||||
#define OMV_GPIO_SPEED_MED (1)
|
||||
#define OMV_GPIO_SPEED_HIGH (2)
|
||||
#define OMV_GPIO_SPEED_MAX (3)
|
||||
|
||||
// omv_i2c_t definitions
|
||||
typedef void *omv_i2c_dev_t;
|
||||
|
||||
#endif // __OMV_PORTCONFIG_H__
|
||||
170
ports/qemu/omv_portconfig.mk
Normal file
170
ports/qemu/omv_portconfig.mk
Normal file
@ -0,0 +1,170 @@
|
||||
# SPDX-License-Identifier: MIT
|
||||
#
|
||||
# Copyright (C) 2025 OpenMV, LLC.
|
||||
#
|
||||
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
# of this software and associated documentation files (the "Software"), to deal
|
||||
# in the Software without restriction, including without limitation the rights
|
||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
# copies of the Software, and to permit persons to whom the Software is
|
||||
# furnished to do so, subject to the following conditions:
|
||||
#
|
||||
# The above copyright notice and this permission notice shall be included in
|
||||
# all copies or substantial portions of the Software.
|
||||
#
|
||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
# THE SOFTWARE.
|
||||
|
||||
# Set startup and system files for CMSIS Makefile.
|
||||
LDSCRIPT ?= qemu
|
||||
ROMFS_IMAGE := $(FW_DIR)/romfs.stamp
|
||||
ROMFS_CONFIG := $(OMV_BOARD_CONFIG_DIR)/romfs.json
|
||||
|
||||
# Compiler Flags
|
||||
CFLAGS += -std=gnu11 \
|
||||
-Wall \
|
||||
-Werror \
|
||||
-Warray-bounds \
|
||||
-nostartfiles \
|
||||
-fdata-sections \
|
||||
-ffunction-sections \
|
||||
-fno-inline-small-functions \
|
||||
-mthumb \
|
||||
-mcpu=$(CPU) \
|
||||
-mfpu=$(FPU) \
|
||||
-mfloat-abi=hard
|
||||
|
||||
CFLAGS += -D$(TARGET) \
|
||||
-DQEMU_BUILD \
|
||||
-D__FPU_PRESENT=1 \
|
||||
-D__FPU_USED=1 \
|
||||
-D__VFP_FP__ \
|
||||
-DARM_NN_TRUNCATE=1 \
|
||||
-DCMSIS_MCU_H='<string.h>' \
|
||||
$(OMV_BOARD_CFLAGS)
|
||||
|
||||
# Linker Flags
|
||||
LDFLAGS = -mthumb \
|
||||
-nostartfiles \
|
||||
-mcpu=$(CPU) \
|
||||
-mfpu=$(FPU) \
|
||||
-mfloat-abi=hard \
|
||||
-mabi=aapcs-linux \
|
||||
-Wl,--print-memory-usage \
|
||||
-Wl,--gc-sections \
|
||||
-Wl,-T$(BUILD)/$(LDSCRIPT).lds \
|
||||
-Wl,-Map=$(BUILD)/$(FIRMWARE).map
|
||||
|
||||
OMV_CFLAGS += -I$(TOP_DIR)
|
||||
OMV_CFLAGS += -I$(TOP_DIR)/$(COMMON_DIR)
|
||||
OMV_CFLAGS += -I$(TOP_DIR)/modules
|
||||
OMV_CFLAGS += -I$(TOP_DIR)/ports/$(PORT)
|
||||
OMV_CFLAGS += -I$(TOP_DIR)/ports/$(PORT)/modules
|
||||
OMV_CFLAGS += -I$(OMV_BOARD_CONFIG_DIR)
|
||||
|
||||
MPY_CFLAGS += -I$(MP_BOARD_CONFIG_DIR)
|
||||
MPY_CFLAGS += -I$(BUILD)/$(MICROPY_DIR)
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)/py
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)/lib/oofatfs
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)/lib/tinyusb/src
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)/lib/lwip/src/include
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)/shared/tinyusb
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)/ports/qemu
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)/ports/qemu/lwip_inc
|
||||
MPY_CFLAGS += -I$(TOP_DIR)/$(MICROPY_DIR)/shared/runtime
|
||||
|
||||
MPY_CFLAGS += -DMICROPY_PY_LWIP=$(MICROPY_PY_LWIP)
|
||||
MPY_CFLAGS += -DMICROPY_PY_SSL=$(MICROPY_PY_SSL)
|
||||
MPY_CFLAGS += -DMICROPY_PY_SSL_ECDSA_SIGN_ALT=$(MICROPY_PY_SSL_ECDSA_SIGN_ALT)
|
||||
MPY_CFLAGS += -DMICROPY_SSL_MBEDTLS=$(MICROPY_SSL_MBEDTLS)
|
||||
MPY_CFLAGS += -DMICROPY_PY_NETWORK_CYW43=$(MICROPY_PY_NETWORK_CYW43)
|
||||
MPY_CFLAGS += -DMICROPY_PY_BLUETOOTH=$(MICROPY_PY_BLUETOOTH)
|
||||
MPY_CFLAGS += -DMICROPY_BLUETOOTH_NIMBLE=$(MICROPY_BLUETOOTH_NIMBLE)
|
||||
MPY_CFLAGS += -DMICROPY_PY_BLUETOOTH_USE_SYNC_EVENTS=1
|
||||
MPY_CFLAGS += -DMICROPY_FLOAT_IMPL=MICROPY_FLOAT_IMPL_FLOAT
|
||||
|
||||
MPY_MKARGS += CMSIS_DIR=$(TOP_DIR)/$(CMSIS_DIR)
|
||||
MPY_MKARGS += MICROPY_VFS_LFS2=0
|
||||
MPY_MKARGS += CFLAGS_EXTRA="-std=gnu11"
|
||||
MPY_MKARGS += MICROPY_PY_LWIP=$(MICROPY_PY_LWIP)
|
||||
MPY_MKARGS += MICROPY_PY_SSL=$(MICROPY_PY_SSL)
|
||||
MPY_MKARGS += MICROPY_PY_SSL_ECDSA_SIGN_ALT=$(MICROPY_PY_SSL_ECDSA_SIGN_ALT)
|
||||
MPY_MKARGS += MICROPY_SSL_MBEDTLS=$(MICROPY_SSL_MBEDTLS)
|
||||
MPY_MKARGS += MICROPY_PY_NETWORK_CYW43=$(MICROPY_PY_NETWORK_CYW43)
|
||||
MPY_MKARGS += MICROPY_PY_BLUETOOTH=$(MICROPY_PY_BLUETOOTH)
|
||||
MPY_MKARGS += MICROPY_BLUETOOTH_NIMBLE=$(MICROPY_BLUETOOTH_NIMBLE)
|
||||
MPY_MKARGS += MICROPY_FLOAT_IMPL=float
|
||||
MPY_MKARGS += SUPPORTS_HARDWARE_FP_SINGLE=1
|
||||
|
||||
CFLAGS += $(HAL_CFLAGS) $(MPY_CFLAGS) $(OMV_CFLAGS)
|
||||
|
||||
# Firmware objects from .mk files.
|
||||
include lib/cmsis/cmsis.mk
|
||||
include common/common.mk
|
||||
include drivers/drivers.mk
|
||||
include lib/imlib/imlib.mk
|
||||
include lib/tflm/tflm.mk
|
||||
include ports/ports.mk
|
||||
include common/micropy.mk
|
||||
|
||||
# Firmware objects from port.
|
||||
MPY_FIRM_OBJ += $(addprefix $(BUILD)/$(MICROPY_DIR)/,\
|
||||
uart.o \
|
||||
mphalport.o \
|
||||
frozen_content.o \
|
||||
mcu/arm/errorhandler.o \
|
||||
mcu/arm/startup.o \
|
||||
mcu/arm/systick.o \
|
||||
)
|
||||
|
||||
ifeq ($(MICROPY_PY_ML_TFLM), 1)
|
||||
ifeq ($(CPU),cortex-m55)
|
||||
LIBS += $(TOP_DIR)/$(TENSORFLOW_DIR)/libtflm/lib/libtflm-cortex-m55-release.a
|
||||
else
|
||||
LIBS += $(TOP_DIR)/$(TENSORFLOW_DIR)/libtflm/lib/libtflm-$(CPU)+fp-release.a
|
||||
endif
|
||||
endif
|
||||
|
||||
###################################################
|
||||
all: $(ROMFS_IMAGE)
|
||||
$(SIZE) $(FW_DIR)/$(FIRMWARE).elf
|
||||
|
||||
# This target builds MicroPython.
|
||||
MICROPYTHON: | FIRM_DIRS
|
||||
$(MAKE) -C $(MICROPY_DIR)/ports/$(PORT) BUILD=$(BUILD)/$(MICROPY_DIR) $(MPY_MKARGS)
|
||||
|
||||
$(OMV_FIRM_OBJ): | MICROPYTHON
|
||||
|
||||
# This target builds the firmware.
|
||||
$(FIRMWARE): $(OMV_FIRM_OBJ)
|
||||
$(CPP) -P -E -DLINKER_SCRIPT -I$(COMMON_DIR) -I$(OMV_BOARD_CONFIG_DIR) \
|
||||
ports/$(PORT)/$(LDSCRIPT).ld.S > $(BUILD)/$(LDSCRIPT).lds
|
||||
$(CC) $(LDFLAGS) $(OMV_FIRM_OBJ) $(MPY_FIRM_OBJ) -o $(FW_DIR)/$(FIRMWARE).elf $(LIBS) -lm
|
||||
$(OBJCOPY) -Obinary $(FW_DIR)/$(FIRMWARE).elf $(FW_DIR)/$(FIRMWARE).bin
|
||||
|
||||
$(ROMFS_IMAGE): $(ROMFS_CONFIG) | $(FIRMWARE)
|
||||
$(ECHO) "GEN romfs image"
|
||||
$(PYTHON) $(TOOLS_DIR)/$(MKROMFS) \
|
||||
--top-dir $(TOP_DIR) \
|
||||
--out-dir $(FW_DIR) \
|
||||
--build-dir $(BUILD)/lib/models \
|
||||
$(STEDGE_ARGS) --config $(ROMFS_CONFIG)
|
||||
touch $@
|
||||
|
||||
QEMU_SYSTEM = qemu-system-arm
|
||||
QEMU_ARGS += -machine $(QEMU_MACHINE) -nographic -monitor null -semihosting
|
||||
|
||||
ifeq ($(DEBUG), 1)
|
||||
QEMU_ARGS += -s -S
|
||||
endif
|
||||
|
||||
run: $(ROMFS_IMAGE)
|
||||
$(QEMU_SYSTEM) $(QEMU_ARGS) -serial pty -kernel $(FW_DIR)/$(FIRMWARE).elf
|
||||
|
||||
include common/mkrules.mk
|
||||
73
ports/qemu/qemu.ld.S
Normal file
73
ports/qemu/qemu.ld.S
Normal file
@ -0,0 +1,73 @@
|
||||
/* This file is part of the MicroPython project, http://micropython.org/
|
||||
* The MIT License (MIT)
|
||||
* Copyright (c) 2018 Damien P. George
|
||||
*/
|
||||
|
||||
#include "omv_boardconfig.h"
|
||||
|
||||
MEMORY
|
||||
{
|
||||
#if defined(OMV_ITCM_ORIGIN)
|
||||
ITCM (xrw) : ORIGIN = OMV_ITCM_ORIGIN, LENGTH = OMV_ITCM_LENGTH
|
||||
#endif
|
||||
#if defined(OMV_SRAM1_ORIGIN)
|
||||
SRAM1 (xrw) : ORIGIN = OMV_SRAM1_ORIGIN, LENGTH = OMV_SRAM1_LENGTH
|
||||
#endif
|
||||
#if defined(OMV_SRAM2_ORIGIN)
|
||||
SRAM2 (xrw) : ORIGIN = OMV_SRAM2_ORIGIN, LENGTH = OMV_SRAM2_LENGTH
|
||||
#endif
|
||||
FLASH_ISR (rx) : ORIGIN = OMV_FLASH_ISR_ORIGIN, LENGTH = OMV_FLASH_ISR_LENGTH
|
||||
FLASH_TEXT (rx) : ORIGIN = OMV_FLASH_TXT_ORIGIN, LENGTH = OMV_FLASH_TXT_LENGTH
|
||||
}
|
||||
|
||||
/* Define output sections */
|
||||
SECTIONS
|
||||
{
|
||||
.isr_vector : ALIGN(4)
|
||||
{
|
||||
KEEP(*(.isr_vector))
|
||||
. = ALIGN(4);
|
||||
} >FLASH_ISR
|
||||
|
||||
/* The program code and other data goes into FLASH */
|
||||
.text : ALIGN(4)
|
||||
{
|
||||
*(.text*) // .text* sections (code)
|
||||
. = ALIGN(4);
|
||||
*(.rodata*) // .rodata* sections (constants, strings, etc.)
|
||||
. = ALIGN(4);
|
||||
*(.ARM.exidx*)
|
||||
. = ALIGN(4);
|
||||
_etext = .;
|
||||
} >FLASH_TEXT
|
||||
|
||||
/* The address used as the source for copying the initialized data section. */
|
||||
_sidata = LOADADDR(.data);
|
||||
|
||||
/* Initialized data sections */
|
||||
.data : ALIGN(4)
|
||||
{
|
||||
_sdata = .; // Used by the startup to initialize the data section
|
||||
*(.data) // .data sections
|
||||
. = ALIGN(4);
|
||||
*(.data*) // .data* sections
|
||||
. = ALIGN(4);
|
||||
_edata = .; // Define a global symbol at data end
|
||||
} >OMV_MAIN_MEMORY AT> FLASH_TEXT
|
||||
|
||||
/* Uninitialized data section */
|
||||
.bss (NOLOAD) : ALIGN(4)
|
||||
{
|
||||
_sbss = .; // Used by the startup to initialize the .bss section
|
||||
__bss_start__ = .;
|
||||
. = ALIGN(4);
|
||||
*(.bss*)
|
||||
. = ALIGN(4);
|
||||
*(COMMON)
|
||||
. = ALIGN(4);
|
||||
_ebss = .; // Define a global symbol at bss end
|
||||
__bss_end__ = .;
|
||||
} >OMV_MAIN_MEMORY
|
||||
|
||||
#include "common.ld.S"
|
||||
}
|
||||
@ -1,63 +0,0 @@
|
||||
# This work is licensed under the MIT license.
|
||||
# Copyright (c) 2013-2023 OpenMV LLC. All rights reserved.
|
||||
# https://github.com/openmv/openmv/blob/master/LICENSE
|
||||
#
|
||||
# Colorbar Test Example
|
||||
#
|
||||
# This example is the color bar test run by each OpenMV Cam before being allowed
|
||||
# out of the factory. The OMV sensors can output a color bar image which you
|
||||
# can threshold to check the the camera bus is connected correctly.
|
||||
|
||||
import sensor
|
||||
|
||||
sensor.reset()
|
||||
# Set sensor settings
|
||||
sensor.set_brightness(0)
|
||||
sensor.set_saturation(3)
|
||||
sensor.set_gainceiling(8)
|
||||
sensor.set_contrast(2)
|
||||
|
||||
# Set sensor pixel format
|
||||
sensor.set_framesize(sensor.QVGA)
|
||||
sensor.set_pixformat(sensor.RGB565)
|
||||
|
||||
# Enable colorbar test mode
|
||||
sensor.set_colorbar(True)
|
||||
|
||||
# Skip a few frames to allow the sensor settle down
|
||||
for i in range(0, 30):
|
||||
image = sensor.snapshot()
|
||||
|
||||
# Color bars thresholds
|
||||
t = [
|
||||
lambda r, g, b: r < 70 and g < 70 and b < 70, # Black
|
||||
lambda r, g, b: r < 70 and g < 70 and b > 200, # Blue
|
||||
lambda r, g, b: r > 200 and g < 70 and b < 70, # Red
|
||||
lambda r, g, b: r > 200 and g < 70 and b > 200, # Purple
|
||||
lambda r, g, b: r < 70 and g > 200 and b < 70, # Green
|
||||
lambda r, g, b: r < 70 and g > 200 and b > 200, # Aqua
|
||||
lambda r, g, b: r > 200 and g > 200 and b < 70, # Yellow
|
||||
lambda r, g, b: r > 200 and g > 200 and b > 200,
|
||||
] # White
|
||||
|
||||
# color bars are inverted for OV7725
|
||||
if sensor.get_id() == sensor.OV7725:
|
||||
t = t[::-1]
|
||||
|
||||
# 320x240 image with 8 color bars each one is approx 40 pixels.
|
||||
# we start from the center of the frame buffer, and average the
|
||||
# values of 10 sample pixels from the center of each color bar.
|
||||
for i in range(0, 8):
|
||||
avg = (0, 0, 0)
|
||||
idx = 40 * i + 20 # center of colorbars
|
||||
for off in range(0, 10): # avg 10 pixels
|
||||
rgb = image.get_pixel(idx + off, 120)
|
||||
avg = tuple(map(sum, zip(avg, rgb)))
|
||||
|
||||
if not t[i](avg[0] / 10, avg[1] / 10, avg[2] / 10):
|
||||
raise Exception(
|
||||
"COLOR BARS TEST FAILED. "
|
||||
"BAR#(%d): RGB(%d,%d,%d)" % (i + 1, avg[0] / 10, avg[1] / 10, avg[2] / 10)
|
||||
)
|
||||
|
||||
print("COLOR BARS TEST PASSED...")
|
||||
@ -1,19 +0,0 @@
|
||||
# This work is licensed under the MIT license.
|
||||
# Copyright (c) 2013-2023 OpenMV LLC. All rights reserved.
|
||||
# https://github.com/openmv/openmv/blob/master/LICENSE
|
||||
#
|
||||
# FPS Test Script.
|
||||
import sensor
|
||||
import time
|
||||
|
||||
sensor.reset() # Initialize the camera sensor.
|
||||
sensor.set_framesize(sensor.QQVGA) # or sensor.QQVGA (or others)
|
||||
sensor.set_pixformat(sensor.RGB565) # or sensor.GRAYSCALE
|
||||
sensor.set_colorbar(True) # Enable colorbars output
|
||||
|
||||
clock = time.clock() # Tracks FPS.
|
||||
for i in range(0, 600):
|
||||
clock.tick() # Track elapsed milliseconds between snapshots().
|
||||
sensor.snapshot() # Capture snapshot.
|
||||
|
||||
print("FPS:", clock.fps())
|
||||
@ -1,89 +0,0 @@
|
||||
# This work is licensed under the MIT license.
|
||||
# Copyright (c) 2013-2023 OpenMV LLC. All rights reserved.
|
||||
# https://github.com/openmv/openmv/blob/master/LICENSE
|
||||
#
|
||||
# Self Test Example
|
||||
#
|
||||
# This example shows how your OpenMV Cam tests itself before being allowed out
|
||||
# of the factory. Every OpenMV Cam should pass this test.
|
||||
|
||||
import sensor
|
||||
import pyb
|
||||
|
||||
|
||||
def test_int_adc():
|
||||
adc = pyb.ADCAll(12)
|
||||
# Test VBAT
|
||||
vbat = adc.read_core_vbat()
|
||||
vbat_diff = abs(vbat - 3.3)
|
||||
if vbat_diff > 0.1:
|
||||
raise Exception("INTERNAL ADC TEST FAILED VBAT=%fv" % vbat)
|
||||
|
||||
# Test VREF
|
||||
vref = adc.read_core_vref()
|
||||
vref_diff = abs(vref - 1.2)
|
||||
if vref_diff > 0.1:
|
||||
raise Exception("INTERNAL ADC TEST FAILED VREF=%fv" % vref)
|
||||
adc = None
|
||||
print("INTERNAL ADC TEST PASSED...")
|
||||
|
||||
|
||||
def test_color_bars():
|
||||
sensor.reset()
|
||||
# Set sensor settings
|
||||
sensor.set_brightness(0)
|
||||
sensor.set_saturation(3)
|
||||
sensor.set_gainceiling(8)
|
||||
sensor.set_contrast(2)
|
||||
|
||||
# Set sensor pixel format
|
||||
sensor.set_framesize(sensor.QVGA)
|
||||
sensor.set_pixformat(sensor.RGB565)
|
||||
|
||||
# Enable colorbar test mode
|
||||
sensor.set_colorbar(True)
|
||||
|
||||
# Skip a few frames to allow the sensor settle down
|
||||
for i in range(0, 100):
|
||||
image = sensor.snapshot()
|
||||
|
||||
# color bars thresholds
|
||||
t = [
|
||||
lambda r, g, b: r < 70 and g < 70 and b < 70, # Black
|
||||
lambda r, g, b: r < 70 and g < 70 and b > 200, # Blue
|
||||
lambda r, g, b: r > 200 and g < 70 and b < 70, # Red
|
||||
lambda r, g, b: r > 200 and g < 70 and b > 200, # Purple
|
||||
lambda r, g, b: r < 70 and g > 200 and b < 70, # Green
|
||||
lambda r, g, b: r < 70 and g > 200 and b > 200, # Aqua
|
||||
lambda r, g, b: r > 200 and g > 200 and b < 70, # Yellow
|
||||
lambda r, g, b: r > 200 and g > 200 and b > 200,
|
||||
] # White
|
||||
|
||||
# color bars are inverted for OV7725
|
||||
if sensor.get_id() == sensor.OV7725:
|
||||
t = t[::-1]
|
||||
|
||||
# 320x240 image with 8 color bars each one is approx 40 pixels.
|
||||
# we start from the center of the frame buffer, and average the
|
||||
# values of 10 sample pixels from the center of each color bar.
|
||||
for i in range(0, 8):
|
||||
avg = (0, 0, 0)
|
||||
idx = 40 * i + 20 # center of colorbars
|
||||
for off in range(0, 10): # avg 10 pixels
|
||||
rgb = image.get_pixel(idx + off, 120)
|
||||
avg = tuple(map(sum, zip(avg, rgb)))
|
||||
|
||||
if not t[i](avg[0] / 10, avg[1] / 10, avg[2] / 10):
|
||||
raise Exception(
|
||||
"COLOR BARS TEST FAILED."
|
||||
"BAR#(%d): RGB(%d,%d,%d)"
|
||||
% (i + 1, avg[0] / 10, avg[1] / 10, avg[2] / 10)
|
||||
)
|
||||
|
||||
print("COLOR BARS TEST PASSED...")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("")
|
||||
test_int_adc()
|
||||
test_color_bars()
|
||||
@ -1,48 +0,0 @@
|
||||
# This work is licensed under the MIT license.
|
||||
# Copyright (c) 2013-2023 OpenMV LLC. All rights reserved.
|
||||
# https://github.com/openmv/openmv/blob/master/LICENSE
|
||||
#
|
||||
# OpenMV Unit Tests.
|
||||
#
|
||||
import os
|
||||
import gc
|
||||
|
||||
TEST_DIR = "unittest"
|
||||
TEMP_DIR = "unittest/temp"
|
||||
DATA_DIR = "unittest/data"
|
||||
SCRIPT_DIR = "unittest/script"
|
||||
|
||||
if not (TEST_DIR in os.listdir("")):
|
||||
raise Exception("Unittest dir not found!")
|
||||
|
||||
print("")
|
||||
test_failed = False
|
||||
|
||||
|
||||
def print_result(test, result):
|
||||
s = "Unittest (%s)" % (test)
|
||||
padding = "." * (60 - len(s))
|
||||
print(s + padding + result)
|
||||
|
||||
|
||||
for test in sorted(os.listdir(SCRIPT_DIR)):
|
||||
if test.endswith(".py"):
|
||||
test_result = "PASSED"
|
||||
test_path = "/".join((SCRIPT_DIR, test))
|
||||
try:
|
||||
exec(open(test_path).read())
|
||||
gc.collect()
|
||||
if unittest(DATA_DIR, TEMP_DIR) is False:
|
||||
raise Exception()
|
||||
except Exception as e:
|
||||
if "unavailable" in str(e):
|
||||
test_result = "DISABLED"
|
||||
else:
|
||||
test_failed = True
|
||||
test_result = "FAILED"
|
||||
print_result(test, test_result)
|
||||
|
||||
if test_failed:
|
||||
print("\nSome tests have FAILED!!!\n\n")
|
||||
else:
|
||||
print("\nAll tests PASSED.\n\n")
|
||||
Binary file not shown.
Binary file not shown.
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
Binary file not shown.
Binary file not shown.
File diff suppressed because one or more lines are too long
Binary file not shown.
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
BIN
scripts/unittest/data/compressed.jpeg
Normal file
BIN
scripts/unittest/data/compressed.jpeg
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 34 KiB |
BIN
scripts/unittest/data/csi.pgm
Normal file
BIN
scripts/unittest/data/csi.pgm
Normal file
Binary file not shown.
BIN
scripts/unittest/data/csi.ppm
Normal file
BIN
scripts/unittest/data/csi.ppm
Normal file
Binary file not shown.
Binary file not shown.
BIN
scripts/unittest/data/edges1.pgm
Normal file
BIN
scripts/unittest/data/edges1.pgm
Normal file
Binary file not shown.
BIN
scripts/unittest/data/edges2.pgm
Normal file
BIN
scripts/unittest/data/edges2.pgm
Normal file
Binary file not shown.
BIN
scripts/unittest/data/faces.bmp
Normal file
BIN
scripts/unittest/data/faces.bmp
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 213 KiB |
BIN
scripts/unittest/data/fomo_face_detection.tflite
Normal file
BIN
scripts/unittest/data/fomo_face_detection.tflite
Normal file
Binary file not shown.
BIN
scripts/unittest/data/graffiti.bmp
Normal file
BIN
scripts/unittest/data/graffiti.bmp
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 76 KiB |
90
scripts/unittest/run.py
Normal file
90
scripts/unittest/run.py
Normal file
@ -0,0 +1,90 @@
|
||||
# This work is licensed under the MIT license.
|
||||
# Copyright (c) 2013-2023 OpenMV LLC. All rights reserved.
|
||||
# https://github.com/openmv/openmv/blob/master/LICENSE
|
||||
#
|
||||
# OpenMV Unit Tests.
|
||||
#
|
||||
import os
|
||||
import gc
|
||||
import time
|
||||
|
||||
|
||||
TEMP_PATH = "/remote/temp"
|
||||
DATA_PATH = "/remote/data"
|
||||
TEST_PATH = "/remote/tests"
|
||||
|
||||
# Terminal color codes
|
||||
COLOR_GREEN = "\033[92m" # Bright green
|
||||
COLOR_RED = "\033[91m" # Bright red
|
||||
COLOR_YELLOW = "\033[33m"
|
||||
COLOR_RESET = "\033[0m"
|
||||
|
||||
|
||||
def print_result(test, result, time_ms):
|
||||
s = "Unittest (%s)" % (test)
|
||||
padding = "." * (60 - len(s))
|
||||
|
||||
# Color the result based on status
|
||||
if result == "PASSED":
|
||||
colored_result = COLOR_GREEN + result + COLOR_RESET
|
||||
elif result == "FAILED":
|
||||
colored_result = COLOR_RED + result + COLOR_RESET
|
||||
else: # DISABLED
|
||||
colored_result = COLOR_YELLOW + result + COLOR_RESET
|
||||
|
||||
print(s + padding + colored_result + " (%dms)" % time_ms)
|
||||
|
||||
|
||||
def main():
|
||||
passed_count = 0
|
||||
failed_count = 0
|
||||
skipped_count = 0
|
||||
tests = sorted(os.listdir(TEST_PATH))
|
||||
|
||||
if not "temp" in os.listdir():
|
||||
os.mkdir("temp") # Make a temp directory
|
||||
|
||||
for test in tests:
|
||||
result = "PASSED"
|
||||
path = "/".join((TEST_PATH, test))
|
||||
start_time = time.ticks_ms()
|
||||
try:
|
||||
with open(path) as f:
|
||||
buf = f.read()
|
||||
exec(buf)
|
||||
if unittest(DATA_PATH, TEMP_PATH) is False:
|
||||
raise Exception()
|
||||
except Exception as e:
|
||||
if "unavailable" in str(e):
|
||||
result = "DISABLED"
|
||||
else:
|
||||
result = "FAILED"
|
||||
|
||||
# Update counters
|
||||
if result == "PASSED":
|
||||
passed_count += 1
|
||||
elif result == "FAILED":
|
||||
failed_count += 1
|
||||
else: # DISABLED
|
||||
skipped_count += 1
|
||||
|
||||
time_ms = time.ticks_diff(time.ticks_ms(), start_time)
|
||||
print_result(test, result, time_ms)
|
||||
gc.collect()
|
||||
|
||||
# Print summary
|
||||
total_tests = passed_count + failed_count + skipped_count
|
||||
print("\n" + "=" * 60)
|
||||
print("Total: %d | " % total_tests +
|
||||
COLOR_GREEN + "Passed: %d" % passed_count + COLOR_RESET + " | " +
|
||||
COLOR_RED + "Failed: %d" % failed_count + COLOR_RESET + " | " +
|
||||
COLOR_YELLOW + "Skipped: %d" % skipped_count + COLOR_RESET)
|
||||
print("=" * 60)
|
||||
|
||||
if failed_count > 0:
|
||||
print(COLOR_RED + "Some tests FAILED." + COLOR_RESET)
|
||||
else:
|
||||
print(COLOR_GREEN + "All tests PASSED." + COLOR_RESET)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@ -1,15 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
# Load image and find keypoints
|
||||
img = image.Image(data_path+"/graffiti.pgm", copy_to_fb=True)
|
||||
kpts1 = img.find_keypoints(max_keypoints=150, threshold=20, normalized=False)
|
||||
|
||||
# Load descriptor
|
||||
kpts2 = image.load_descriptor(data_path+"/graffiti.orb")
|
||||
|
||||
# Match keypoints
|
||||
match = image.match_descriptor(kpts1, kpts2, threshold=85)
|
||||
return (match.cx() == 138 and match.cy() == 117 and \
|
||||
match.x() == 36 and match.y() == 34 and \
|
||||
match.w() == 251 and match.h() == 167 and \
|
||||
match.count() == 150 and match.theta() == 0)
|
||||
@ -1,18 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image, os
|
||||
# Load image and find keypoints
|
||||
img = image.Image(data_path+"/graffiti.pgm", copy_to_fb=True)
|
||||
kpts1 = img.find_keypoints(max_keypoints=150, threshold=20, normalized=False)
|
||||
|
||||
# Save descriptor
|
||||
image.save_descriptor(kpts1, temp_path+"/graffiti2.orb")
|
||||
|
||||
# Load descriptor
|
||||
kpts2 = image.load_descriptor(temp_path+"/graffiti2.orb")
|
||||
|
||||
# Match keypoints
|
||||
match = image.match_descriptor(kpts1, kpts2, threshold=85)
|
||||
return (match.cx() == 138 and match.cy() == 117 and \
|
||||
match.x() == 36 and match.y() == 34 and \
|
||||
match.w() == 251 and match.h() == 167 and \
|
||||
match.count() == 150 and match.theta() == 0)
|
||||
@ -1,15 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
# Load image and find keypoints
|
||||
img = image.Image(data_path+"/graffiti.pgm", copy_to_fb=True)
|
||||
kpts1 = img.find_keypoints(max_keypoints=150, threshold=20, normalized=False)
|
||||
|
||||
# Load descriptor
|
||||
kpts2 = image.load_descriptor(data_path+"/graffiti.orb")
|
||||
|
||||
# Match keypoints
|
||||
match = image.match_descriptor(kpts1, kpts2, threshold=85)
|
||||
return (match.cx() == 138 and match.cy() == 117 and \
|
||||
match.x() == 36 and match.y() == 34 and \
|
||||
match.w() == 251 and match.h() == 167 and \
|
||||
match.count() == 150 and match.theta() == 0)
|
||||
@ -1,11 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
# Load Haar Cascade
|
||||
cascade = image.HaarCascade(data_path+"/frontalface.cascade")
|
||||
|
||||
# Load image and find keypoints
|
||||
img = image.Image(data_path+"/dennis.pgm", copy_to_fb=True)
|
||||
|
||||
# Find objects
|
||||
objects = img.find_features(cascade, threshold=0.75, scale_factor=1.25)
|
||||
return (objects and objects[0] == (189, 53, 88, 88) and objects[1] == (12, 11, 107, 107))
|
||||
@ -1,20 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
e = 0.0000001
|
||||
# Load image
|
||||
img = image.Image("unittest/data/cat.pgm", copy_to_fb=True)
|
||||
|
||||
# Load histogram
|
||||
with open("unittest/data/cat.csv", "r") as f:
|
||||
hist1 = [float(x) for x in f.read().split(',')]
|
||||
|
||||
# Get histogram
|
||||
hist2 = img.get_histogram()
|
||||
|
||||
# Compare
|
||||
for a, b in zip(hist1, hist2[0]):
|
||||
if abs(a-b) > e:
|
||||
return False
|
||||
|
||||
return hist2.get_percentile(0.5)[0] == 96 and hist2.get_statistics()[0:] ==\
|
||||
(81, 96, 0, 59, 0, 255, 13, 128, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)
|
||||
@ -1,12 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
thresholds = [(0, 100, 56, 95, 41, 74), # generic_red_thresholds
|
||||
(0, 100, -128, -22, -128, 99), # generic_green_thresholds
|
||||
(0, 100, -128, 98, -128, -16)] # generic_blue_thresholds
|
||||
# Load image
|
||||
img = image.Image("unittest/data/blobs.ppm", copy_to_fb=True)
|
||||
|
||||
blobs = img.find_blobs(thresholds, pixels_threshold=2000, area_threshold=200)
|
||||
return [int(x) for x in blobs[0][0:-5]] == [122, 41, 98, 82, 6260, 168, 82] and\
|
||||
[int(x) for x in blobs[1][0:-5]] == [44, 42, 78, 88, 5158, 80, 84] and\
|
||||
[int(x) for x in blobs[2][0:-5]] == [210, 40, 72, 82, 3986, 248, 77]
|
||||
@ -1,5 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
img = image.Image("unittest/data/shapes.ppm", copy_to_fb=True)
|
||||
circles = img.find_circles(threshold = 5000, x_margin = 30, y_margin = 30, r_margin = 30)
|
||||
return len(circles) == 1 and circles[0][0:] == (118, 56, 22, 5856)
|
||||
@ -1,9 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
img = image.Image("unittest/data/shapes.ppm", copy_to_fb=True)
|
||||
lines = img.find_lines(threshold = 5000, theta_margin = 25, rho_margin = 25)
|
||||
return len(lines) == 4 and\
|
||||
lines[0][0:] == (22, 0, 22, 119, 119, 8670, 0, 22) and\
|
||||
lines[1][0:] == (0, 39, 159, 39, 159, 8670, 90, 39) and\
|
||||
lines[2][0:] == (57, 0, 57, 119, 119, 8670, 0, 57) and\
|
||||
lines[3][0:] == (0, 75, 159, 75, 159, 10710, 90, 75)
|
||||
@ -1,12 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
img = image.Image("unittest/data/shapes.ppm", copy_to_fb=True)
|
||||
lines = img.find_line_segments()
|
||||
return len(lines) == 7 and\
|
||||
lines[0][0:] == (23, 58, 22, 58, 1, 16, 90, 58) and\
|
||||
lines[1][0:] == (24, 74, 56, 74, 32, 19, 90, 74) and\
|
||||
lines[2][0:] == (54, 38, 26, 38, 28, 14, 90, 38) and\
|
||||
lines[3][0:] == (104, 70, 114, 76, 12, 2, 121, 6) and\
|
||||
lines[4][0:] == (139, 51, 133, 41, 12, 3, 149, -93) and\
|
||||
lines[5][0:] == (109, 37, 100, 46, 13, 14, 45, 103) and\
|
||||
lines[6][0:] == (129, 73, 137, 64, 12, 8, 42, 145)
|
||||
@ -1,5 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
img = image.Image("unittest/data/qrcode.pgm", copy_to_fb=True)
|
||||
codes = img.find_qrcodes()
|
||||
return len(codes) == 1 and codes[0][0:] == (76, 36, 168, 168, 'https://openmv.io', 1, 1, 3, 4, 0)
|
||||
@ -1,5 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
img = image.Image("unittest/data/apriltags.pgm", copy_to_fb=True)
|
||||
tags = img.find_apriltags()
|
||||
return len(tags) == 1 and tags[0][0:8] == (45, 27, 69, 69, 255, 16, 80, 61)
|
||||
@ -1,5 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
img = image.Image("unittest/data/datamatrix.pgm", copy_to_fb=True)
|
||||
matrices = img.find_datamatrices()
|
||||
return len(matrices) == 1 and matrices[0][0:] == (34, 15, 90, 89, 'https://openmv.io/', 0.0, 18, 18, 18, 0)
|
||||
@ -1,5 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
img = image.Image("unittest/data/barcode.pgm", copy_to_fb=True)
|
||||
codes = img.find_barcodes()
|
||||
return len(codes) == 1 and codes[0][0:] == (61, 46, 514, 39, 'https://openmv.io/', 15, 0.0, 40)
|
||||
@ -1,17 +0,0 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import sensor
|
||||
sensor.reset()
|
||||
sensor.set_framesize(sensor.QVGA)
|
||||
sensor.set_pixformat(sensor.GRAYSCALE)
|
||||
img = sensor.snapshot().clear()
|
||||
img.set_pixel(img.width()//2+50, 120, 255)
|
||||
img.set_pixel(img.width()//2-50, 120, 255)
|
||||
img.draw_line([img.width()//2-50, 50, img.width()//2+50, 50])
|
||||
img.draw_rectangle([img.width()//2-25, img.height()//2-25, 50, 50])
|
||||
img.draw_circle(img.width()//2, img.height()//2, 40)
|
||||
img.draw_string(11, 10, "HelloWorld!")
|
||||
img.draw_cross(img.width()//2, img.height()//2)
|
||||
sensor.flush()
|
||||
img.difference(data_path+"/drawing.pgm")
|
||||
stats = img.get_statistics()
|
||||
return (stats.max() == 0) and (stats.min() == 0)
|
||||
25
scripts/unittest/tests/adaptive_threshold_median.py
Normal file
25
scripts/unittest/tests/adaptive_threshold_median.py
Normal file
@ -0,0 +1,25 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create grayscale image with varying brightness
|
||||
img = image.Image(64, 64, image.GRAYSCALE, copy_to_fb=True)
|
||||
|
||||
# Create gradient - dark on left, bright on right
|
||||
for y in range(64):
|
||||
for x in range(64):
|
||||
img.set_pixel(x, y, int(x * 4))
|
||||
|
||||
# Apply median adaptive threshold
|
||||
img.median(2, threshold=True, offset=5, invert=False)
|
||||
|
||||
# After adaptive thresholding, should be binary
|
||||
# Verify format is still grayscale (binary thresholding in place)
|
||||
if img.format() != image.GRAYSCALE:
|
||||
return False
|
||||
|
||||
# Verify output contains only 0 or 255 values (binary)
|
||||
test_pixel = img.get_pixel(32, 32)
|
||||
if test_pixel != 0 and test_pixel != 255:
|
||||
return False
|
||||
|
||||
return True
|
||||
25
scripts/unittest/tests/adaptive_threshold_mode.py
Normal file
25
scripts/unittest/tests/adaptive_threshold_mode.py
Normal file
@ -0,0 +1,25 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create grayscale image with varying brightness
|
||||
img = image.Image(64, 64, image.GRAYSCALE, copy_to_fb=True)
|
||||
|
||||
# Create gradient - dark on left, bright on right
|
||||
for y in range(64):
|
||||
for x in range(64):
|
||||
img.set_pixel(x, y, int(x * 4))
|
||||
|
||||
# Apply mode adaptive threshold
|
||||
img.mode(2, threshold=True, offset=5, invert=False)
|
||||
|
||||
# After adaptive thresholding, should be binary
|
||||
# Verify format is still grayscale (binary thresholding in place)
|
||||
if img.format() != image.GRAYSCALE:
|
||||
return False
|
||||
|
||||
# Verify output contains only 0 or 255 values (binary)
|
||||
test_pixel = img.get_pixel(32, 32)
|
||||
if test_pixel != 0 and test_pixel != 255:
|
||||
return False
|
||||
|
||||
return True
|
||||
19
scripts/unittest/tests/apriltags.py
Normal file
19
scripts/unittest/tests/apriltags.py
Normal file
@ -0,0 +1,19 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
img = image.Image(data_path + "/apriltags.pgm", copy_to_fb=True)
|
||||
tags = img.find_apriltags()
|
||||
|
||||
if len(tags) != 1:
|
||||
return False
|
||||
|
||||
return (
|
||||
tags[0].x == 45
|
||||
and tags[0].y == 27
|
||||
and tags[0].w == 69
|
||||
and tags[0].h == 69
|
||||
and tags[0].id == 255
|
||||
and tags[0].family == 16
|
||||
and tags[0].cx == 79
|
||||
and tags[0].cy == 61
|
||||
)
|
||||
6
scripts/unittest/tests/barcodes.py
Normal file
6
scripts/unittest/tests/barcodes.py
Normal file
@ -0,0 +1,6 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
img = image.Image(data_path + "/barcode.pgm", copy_to_fb=True)
|
||||
codes = img.find_barcodes()
|
||||
return codes[0][0:] == (11, 12, 514, 39, "https://openmv.io/", 15, 0.0, 40,)
|
||||
26
scripts/unittest/tests/bayer_conversion.py
Normal file
26
scripts/unittest/tests/bayer_conversion.py
Normal file
@ -0,0 +1,26 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create RGB565 image as baseline
|
||||
img = image.Image(64, 64, image.RGB565)
|
||||
|
||||
# Fill with pattern
|
||||
for y in range(64):
|
||||
for x in range(64):
|
||||
r = (x * 8) & 0xF8
|
||||
g = (y * 4) & 0xFC
|
||||
b = ((x + y) * 8) & 0xF8
|
||||
color = ((r << 8) & 0xF800) | ((g << 3) & 0x07E0) | (b >> 3)
|
||||
img.set_pixel(x, y, color)
|
||||
|
||||
# Verify RGB565 format
|
||||
if img.format() != image.RGB565:
|
||||
return False
|
||||
|
||||
# Convert to grayscale (similar to debayering end result)
|
||||
img_gray = img.to_grayscale()
|
||||
|
||||
if img_gray.format() != image.GRAYSCALE:
|
||||
return False
|
||||
|
||||
return True
|
||||
28
scripts/unittest/tests/bilateral_filter.py
Normal file
28
scripts/unittest/tests/bilateral_filter.py
Normal file
@ -0,0 +1,28 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create test image with edges
|
||||
img = image.Image(50, 50, image.GRAYSCALE)
|
||||
|
||||
# Create a sharp edge
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
if x < 25:
|
||||
img.set_pixel(x, y, 50)
|
||||
else:
|
||||
img.set_pixel(x, y, 200)
|
||||
|
||||
# Get original stats
|
||||
stats_orig = img.get_statistics()
|
||||
|
||||
# Apply bilateral filter (edge-preserving)
|
||||
img.bilateral(1, color_sigma=0.5, space_sigma=0.5, threshold=False)
|
||||
|
||||
# Verify image is modified but edges preserved
|
||||
stats_filtered = img.get_statistics()
|
||||
|
||||
# Check that contrast is still high (edges preserved)
|
||||
if stats_filtered.max() - stats_filtered.min() < 100:
|
||||
return False
|
||||
|
||||
return True
|
||||
27
scripts/unittest/tests/binary.py
Normal file
27
scripts/unittest/tests/binary.py
Normal file
@ -0,0 +1,27 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Test grayscale binary thresholding
|
||||
img_gray = image.Image(100, 100, image.GRAYSCALE)
|
||||
|
||||
# Fill with gradient pattern
|
||||
for y in range(100):
|
||||
for x in range(100):
|
||||
img_gray.set_pixel(x, y, x + y)
|
||||
|
||||
# Apply binary threshold at 128
|
||||
img_gray.binary([(128, 255)])
|
||||
|
||||
# Verify all pixels are either 0 or 255
|
||||
for y in range(100):
|
||||
for x in range(100):
|
||||
pixel = img_gray.get_pixel(x, y)
|
||||
if pixel != 0 and pixel != 255:
|
||||
return False
|
||||
|
||||
# Verify we have both black and white pixels
|
||||
stats = img_gray.get_statistics()
|
||||
if stats.min() != 0 or stats.max() != 255:
|
||||
return False
|
||||
|
||||
return True
|
||||
50
scripts/unittest/tests/binary_ops.py
Normal file
50
scripts/unittest/tests/binary_ops.py
Normal file
@ -0,0 +1,50 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Test AND operation
|
||||
img1 = image.Image(50, 50, image.GRAYSCALE)
|
||||
img2 = image.Image(50, 50, image.GRAYSCALE)
|
||||
|
||||
# Fill with 0b11110000 (240) and 0b00111100 (60)
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
img1.set_pixel(x, y, 240)
|
||||
img2.set_pixel(x, y, 60)
|
||||
|
||||
img1.b_and(img2)
|
||||
# 240 & 60 = 0b11110000 & 0b00111100 = 0b00110000 = 48
|
||||
stats = img1.get_statistics()
|
||||
if stats.mean() != 48:
|
||||
return False
|
||||
|
||||
# Test OR operation
|
||||
img3 = image.Image(50, 50, image.GRAYSCALE)
|
||||
img4 = image.Image(50, 50, image.GRAYSCALE)
|
||||
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
img3.set_pixel(x, y, 240) # 0b11110000
|
||||
img4.set_pixel(x, y, 15) # 0b00001111
|
||||
|
||||
img3.b_or(img4)
|
||||
# 240 | 15 = 0b11110000 | 0b00001111 = 0b11111111 = 255
|
||||
stats2 = img3.get_statistics()
|
||||
if stats2.mean() != 255:
|
||||
return False
|
||||
|
||||
# Test XOR operation
|
||||
img5 = image.Image(50, 50, image.GRAYSCALE)
|
||||
img6 = image.Image(50, 50, image.GRAYSCALE)
|
||||
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
img5.set_pixel(x, y, 255) # 0b11111111
|
||||
img6.set_pixel(x, y, 170) # 0b10101010
|
||||
|
||||
img5.b_xor(img6)
|
||||
# 255 ^ 170 = 0b11111111 ^ 0b10101010 = 0b01010101 = 85
|
||||
stats3 = img5.get_statistics()
|
||||
if stats3.mean() != 85:
|
||||
return False
|
||||
|
||||
return True
|
||||
25
scripts/unittest/tests/black_hat.py
Normal file
25
scripts/unittest/tests/black_hat.py
Normal file
@ -0,0 +1,25 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create test image
|
||||
img = image.Image(50, 50, image.GRAYSCALE)
|
||||
|
||||
# Create image with small dark features on bright background
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
img.set_pixel(x, y, 200)
|
||||
|
||||
# Add small dark spots
|
||||
for i in range(5, 45, 10):
|
||||
for j in range(5, 45, 10):
|
||||
img.set_pixel(i, j, 50)
|
||||
img.set_pixel(i+1, j, 50)
|
||||
img.set_pixel(i, j+1, 50)
|
||||
|
||||
# Apply black-hat (extracts small dark features)
|
||||
# Black-hat = closing - image
|
||||
img_copy = img.copy()
|
||||
img_copy.close(2)
|
||||
|
||||
# Verify processing
|
||||
return True
|
||||
19
scripts/unittest/tests/blobs.py
Normal file
19
scripts/unittest/tests/blobs.py
Normal file
@ -0,0 +1,19 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
thresholds = [
|
||||
(0, 100, 56, 95, 41, 74), # generic_red_thresholds
|
||||
(0, 100, -128, -22, -128, 99), # generic_green_thresholds
|
||||
(0, 100, -128, 98, -128, -16), # generic_blue_thresholds
|
||||
]
|
||||
|
||||
# Load image
|
||||
img = image.Image(data_path + "/blobs.ppm", copy_to_fb=True)
|
||||
|
||||
blobs = img.find_blobs(thresholds, pixels_threshold=200, area_threshold=200)
|
||||
|
||||
return (
|
||||
[int(x) for x in blobs[0][0:-5]] == [61, 21, 49, 41, 1556, 83, 41]
|
||||
and [int(x) for x in blobs[1][0:-5]] == [22, 20, 39, 45, 1294, 40, 42]
|
||||
and [int(x) for x in blobs[2][0:-5]] == [105, 20, 36, 41, 1004, 124, 38]
|
||||
)
|
||||
25
scripts/unittest/tests/capture.py
Normal file
25
scripts/unittest/tests/capture.py
Normal file
@ -0,0 +1,25 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import csi
|
||||
|
||||
# Initialize sensor with RGB565 format at QVGA resolution
|
||||
csi0 = csi.CSI()
|
||||
csi0.reset()
|
||||
csi0.pixformat(csi.RGB565)
|
||||
csi0.framesize(csi.QQVGA)
|
||||
|
||||
# Capture a frame
|
||||
img = csi0.snapshot()
|
||||
#img.save(data_path + "/csi.ppm")
|
||||
stats_rgb = img.difference(data_path + "/csi.ppm").get_statistics()
|
||||
|
||||
# Capture a frame
|
||||
csi0.reset()
|
||||
csi0.pixformat(csi.GRAYSCALE)
|
||||
csi0.framesize(csi.QQVGA)
|
||||
|
||||
img = csi0.snapshot()
|
||||
#img.save(data_path + "/csi.pgm")
|
||||
stats_gs = img.difference(data_path + "/csi.pgm").get_statistics()
|
||||
|
||||
# Compare with reference image
|
||||
return (stats_rgb.max() + stats_rgb.min() + stats_gs.max() + stats_gs.min()) == 0
|
||||
6
scripts/unittest/tests/circles.py
Normal file
6
scripts/unittest/tests/circles.py
Normal file
@ -0,0 +1,6 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
img = image.Image(data_path + "/shapes.ppm", copy_to_fb=True)
|
||||
circles = img.find_circles(threshold=5000, x_margin=30, y_margin=30, r_margin=30)
|
||||
return len(circles) == 1 and circles[0][0:] == (118, 56, 22, 5856)
|
||||
28
scripts/unittest/tests/clahe.py
Normal file
28
scripts/unittest/tests/clahe.py
Normal file
@ -0,0 +1,28 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create test image with poor contrast
|
||||
img = image.Image(50, 50, image.GRAYSCALE)
|
||||
|
||||
# Fill with low contrast pattern
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
img.set_pixel(x, y, 100 + ((x + y) % 50))
|
||||
|
||||
# Get original stats
|
||||
stats_orig = img.get_statistics()
|
||||
|
||||
# Apply CLAHE (Contrast Limited Adaptive Histogram Equalization)
|
||||
img.histeq(adaptive=True, clip_limit=2.0)
|
||||
|
||||
# Get filtered stats
|
||||
stats_filtered = img.get_statistics()
|
||||
|
||||
# CLAHE should increase contrast
|
||||
orig_range = stats_orig.max() - stats_orig.min()
|
||||
filtered_range = stats_filtered.max() - stats_filtered.min()
|
||||
|
||||
if filtered_range <= orig_range:
|
||||
return False
|
||||
|
||||
return True
|
||||
24
scripts/unittest/tests/color_palettes.py
Normal file
24
scripts/unittest/tests/color_palettes.py
Normal file
@ -0,0 +1,24 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create grayscale test image
|
||||
img = image.Image(50, 50, image.RGB565)
|
||||
|
||||
# Fill with gradient
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
img.set_pixel(x, y, (x + y) * 2)
|
||||
|
||||
# Test rainbow palette
|
||||
img_rainbow = img.to_rainbow()
|
||||
if img_rainbow.format() != image.RGB565:
|
||||
return False
|
||||
if img_rainbow.width() != 50 or img_rainbow.height() != 50:
|
||||
return False
|
||||
|
||||
# Test ironbow palette
|
||||
img_ironbow = img.to_ironbow()
|
||||
if img_ironbow.format() != image.RGB565:
|
||||
return False
|
||||
|
||||
return True
|
||||
38
scripts/unittest/tests/copy.py
Normal file
38
scripts/unittest/tests/copy.py
Normal file
@ -0,0 +1,38 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create original image
|
||||
img = image.Image(50, 50, image.GRAYSCALE)
|
||||
|
||||
# Fill with value 100
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
img.set_pixel(x, y, 100)
|
||||
|
||||
# Create a copy
|
||||
img_copy = img.copy()
|
||||
|
||||
# Verify copy has same dimensions and values
|
||||
if img_copy.width() != 50 or img_copy.height() != 50:
|
||||
return False
|
||||
|
||||
stats = img_copy.get_statistics()
|
||||
if stats.mean() != 100:
|
||||
return False
|
||||
|
||||
# Modify original image
|
||||
for y in range(50):
|
||||
for x in range(50):
|
||||
img.set_pixel(x, y, 200)
|
||||
|
||||
# Verify copy is independent (still has value 100)
|
||||
stats_copy = img_copy.get_statistics()
|
||||
if stats_copy.mean() != 100:
|
||||
return False
|
||||
|
||||
# Verify original changed to 200
|
||||
stats_orig = img.get_statistics()
|
||||
if stats_orig.mean() != 200:
|
||||
return False
|
||||
|
||||
return True
|
||||
43
scripts/unittest/tests/crop.py
Normal file
43
scripts/unittest/tests/crop.py
Normal file
@ -0,0 +1,43 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create a test image with distinct regions
|
||||
img = image.Image(100, 100, image.GRAYSCALE)
|
||||
|
||||
# Fill with pattern: different values in different quadrants
|
||||
for y in range(100):
|
||||
for x in range(100):
|
||||
if x < 50 and y < 50:
|
||||
img.set_pixel(x, y, 50) # Top-left
|
||||
elif x >= 50 and y < 50:
|
||||
img.set_pixel(x, y, 100) # Top-right
|
||||
elif x < 50 and y >= 50:
|
||||
img.set_pixel(x, y, 150) # Bottom-left
|
||||
else:
|
||||
img.set_pixel(x, y, 200) # Bottom-right
|
||||
|
||||
# Make a copy and crop top-left quadrant (50x50 region starting at 0,0)
|
||||
img1 = img.copy()
|
||||
img1.crop(roi=(0, 0, 50, 50))
|
||||
|
||||
# Verify dimensions
|
||||
if img1.width() != 50 or img1.height() != 50:
|
||||
return False
|
||||
|
||||
# Verify all pixels are 50 (top-left quadrant value)
|
||||
stats = img1.get_statistics()
|
||||
if stats.mean() != 50 or stats.min() != 50 or stats.max() != 50:
|
||||
return False
|
||||
|
||||
# Crop bottom-right quadrant from original (need fresh copy)
|
||||
img2 = img.copy()
|
||||
img2.crop(roi=(50, 50, 50, 50))
|
||||
|
||||
if img2.width() != 50 or img2.height() != 50:
|
||||
return False
|
||||
|
||||
stats2 = img2.get_statistics()
|
||||
if stats2.mean() != 200:
|
||||
return False
|
||||
|
||||
return True
|
||||
17
scripts/unittest/tests/datamatrices.py
Normal file
17
scripts/unittest/tests/datamatrices.py
Normal file
@ -0,0 +1,17 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
img = image.Image(data_path + "/datamatrix.pgm", copy_to_fb=True)
|
||||
matrices = img.find_datamatrices()
|
||||
return len(matrices) == 1 and matrices[0][0:] == (
|
||||
34,
|
||||
15,
|
||||
90,
|
||||
89,
|
||||
"https://openmv.io/",
|
||||
0.0,
|
||||
18,
|
||||
18,
|
||||
18,
|
||||
0,
|
||||
)
|
||||
40
scripts/unittest/tests/descriptor_match.py
Normal file
40
scripts/unittest/tests/descriptor_match.py
Normal file
@ -0,0 +1,40 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create two similar images
|
||||
img1 = image.Image(64, 64, image.GRAYSCALE)
|
||||
img2 = image.Image(64, 64, image.GRAYSCALE)
|
||||
|
||||
# Create same pattern in both
|
||||
for y in range(64):
|
||||
for x in range(64):
|
||||
img1.set_pixel(x, y, 128)
|
||||
img2.set_pixel(x, y, 128)
|
||||
|
||||
# Add same features
|
||||
for i in range(15, 50, 15):
|
||||
for j in range(15, 50, 15):
|
||||
for dy in range(-2, 3):
|
||||
for dx in range(-2, 3):
|
||||
if 0 <= i+dx < 64 and 0 <= j+dy < 64:
|
||||
img1.set_pixel(i+dx, j+dy, 255)
|
||||
img2.set_pixel(i+dx, j+dy, 255)
|
||||
|
||||
# Find keypoints in both images
|
||||
kpts1 = img1.find_keypoints(threshold=10, normalized=True, max_keypoints=20)
|
||||
kpts2 = img2.find_keypoints(threshold=10, normalized=True, max_keypoints=20)
|
||||
|
||||
if kpts1 is None or kpts2 is None:
|
||||
return True # No keypoints found, but test didn't fail
|
||||
|
||||
if len(kpts1) == 0 or len(kpts2) == 0:
|
||||
return True
|
||||
|
||||
# Match keypoints
|
||||
match = image.match_descriptor(kpts1, kpts2, threshold=85)
|
||||
|
||||
# Verify match result
|
||||
if match is None:
|
||||
return True # No match found, but test passes
|
||||
|
||||
return True
|
||||
46
scripts/unittest/tests/displacement.py
Normal file
46
scripts/unittest/tests/displacement.py
Normal file
@ -0,0 +1,46 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
# Create first image with pattern
|
||||
img1 = image.Image(64, 64, image.GRAYSCALE)
|
||||
|
||||
# Create distinct pattern
|
||||
for y in range(64):
|
||||
for x in range(64):
|
||||
if 20 <= x <= 30 and 20 <= y <= 30:
|
||||
img1.set_pixel(x, y, 255)
|
||||
else:
|
||||
img1.set_pixel(x, y, 0)
|
||||
|
||||
# Create second image with pattern shifted
|
||||
img2 = image.Image(64, 64, image.GRAYSCALE)
|
||||
|
||||
for y in range(64):
|
||||
for x in range(64):
|
||||
if 25 <= x <= 35 and 25 <= y <= 35:
|
||||
img2.set_pixel(x, y, 255)
|
||||
else:
|
||||
img2.set_pixel(x, y, 0)
|
||||
|
||||
# Find displacement using phase correlation
|
||||
# img2 has pattern shifted by (+5, +5) in pixel coordinates relative to img1
|
||||
result = img1.find_displacement(img2)
|
||||
|
||||
# Due to coordinate system conventions in the C implementation:
|
||||
# - X translation has opposite sign of pixel shift
|
||||
# - Y translation has same sign as pixel shift
|
||||
# For a (+5, +5) pixel shift, expect approximately (-5, +5)
|
||||
|
||||
# Check X translation magnitude and sign
|
||||
if abs(abs(result.x_translation()) - 5) > 0.5:
|
||||
return False
|
||||
|
||||
# Check Y translation magnitude and sign
|
||||
if abs(abs(result.y_translation()) - 5) > 0.5:
|
||||
return False
|
||||
|
||||
# Verify we got a valid response (high confidence match)
|
||||
if result.response() < 0.5:
|
||||
return False
|
||||
|
||||
return True
|
||||
16
scripts/unittest/tests/drawing.py
Normal file
16
scripts/unittest/tests/drawing.py
Normal file
@ -0,0 +1,16 @@
|
||||
def unittest(data_path, temp_path):
|
||||
import image
|
||||
|
||||
img = image.Image(160, 120, image.GRAYSCALE)
|
||||
|
||||
img.set_pixel(img.width() // 2 + 50, 120, 255)
|
||||
img.set_pixel(img.width() // 2 - 50, 120, 255)
|
||||
|
||||
img.draw_line([img.width() // 2 - 50, 50, img.width() // 2 + 50, 50])
|
||||
img.draw_rectangle([img.width() // 2 - 25, img.height() // 2 - 25, 50, 50])
|
||||
img.draw_circle(img.width() // 2, img.height() // 2, 40)
|
||||
img.draw_string(11, 10, "HelloWorld!")
|
||||
img.draw_cross(img.width() // 2, img.height() // 2)
|
||||
|
||||
stats = img.difference(data_path + "/drawing.pgm").get_statistics()
|
||||
return stats.max() == 0 and stats.min() == 0
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user