|
rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2', batch_size=16)
上面的这个batch_size的数量只会被加载一次,也就是只会量化batch_size数量这么多的图片
我看rknn.py文件中有个rknn_batch_size,这个函数是不是把batch_size加载进来的图片分rknn_batch_size次量化完,然后就执行rknn.export_rknn了?
我把batch_size=600,再往上加电脑就崩溃了,16g内存扛不住
|
|