lichen/ieemoo-ai-contrast
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

更改

Browse Source
This commit is contained in:
lee
2025-06-19 17:36:24 +08:00
parent bf9604ec29
commit 061820c34f
4 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tools/model_rknn_transform.py
View File

@ -99,7 +99,8 @@ if __name__ == '__main__':
target_platform='rk3588',
model_pruning=False,
compress_weight=False,
single_core_mode=True)
single_core_mode=True,
enable_flash_attention=True)
# rknn.config(
# mean_values=[[127.5, 127.5, 127.5]], # 对于单通道图像，可以设置为 [[127.5]]
# std_values=[[127.5, 127.5, 127.5]], # 对于单通道图像，可以设置为 [[127.5]]
@ -121,7 +122,9 @@ if __name__ == '__main__':
# Build model
print('--> Building model')
ret = rknn.build(do_quantization=True, dataset='./dataset.txt')
ret = rknn.build(do_quantization=True,
dataset='./dataset.txt',
rknn_batch_size=conf['models']['rknn_batch_size'])
# ret = rknn.build(do_quantization=False, dataset='./dataset.txt')
if ret != 0:
print('Build model failed!')