如何解决除了作为字符串分隔符之外,@ in split'@'中还有什么作用?
我正在使用Coral开发板来加速AI模型。
我不明白'@'
是什么意思。
split
使用指定的“分隔符”返回字符串中所有单词的列表。但是我的模型文件名没有'@'。
似乎正在为模型文件分配委托。
模型名称= mobilenet_v2_1.0_224_quant_edgetpu.tflite
import argparse
import time
from PIL import Image
import classify
import tflite_runtime.interpreter as tflite
import platform
EDGETPU_SHARED_LIB = {
'Linux': 'libedgetpu.so.1','Darwin': 'libedgetpu.1.dylib','Windows': 'edgetpu.dll'
}[platform.system()]
def make_interpreter(model_file):
model_file,*device = model_file.split('@')
return tflite.Interpreter(
model_path=model_file,experimental_delegates=[
tflite.load_delegate(EDGETPU_SHARED_LIB,{'device': device[0]} if device else {})
])
谢谢
解决方法
string.split()
的参数只是一个分隔符。它什么也没做。如果分隔符未出现在字符串中,则返回单例列表:[string]
。
model_file,*device = model_file.split('@')
期望model_file.split('@')
返回一个列表,并将该列表的第一个元素分配给model_file
,并将所有后续元素分配给device
(这就是list-unpacking operator *
的作用在这种情况下)。
如果在这种情况下,model_file.split('@')
将返回仅包含一个元素的列表,那么在执行此行之后,device
将是一个空列表[]
。
道歉,我是Google-coral团队的Nam,道歉来自stackoverflow用户,而不是我们。您确实有一个可靠的问题,我第二次回答@Green Cloak Guy,但是为了进一步扩展它:
在我们的有关通过tflite API使用多个tpu的文档中,您可以指定要在哪个设备上加载此模型:https://coral.ai/docs/edgetpu/multiple-edgetpu/#using-the-tensorflow-lite-python-api 基本上,如果您有2个pcie设备和2个USB设备,它将由tflite表示,如下所示:
pci:0
pci:1
usb:0
usb:1
我想这部分没有很好的文档记录,需要将@Green关于python的答案和我们的文档放在一起。但是,在运行演示时,您不仅可以提供模型路径,还可以附加希望运行该模型的设备,例如:
python3 classify_image.py \
--model models/mobilenet_v2_1.0_224_inat_bird_quant_edgetpu.tflite@pci:0 \
--labels models/inat_bird_labels.txt \
--input images/parrot.jpg
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。