TPU学习记录以及遇到的坑

一、错误记录
1、
Traceback (most recent call last):
File "demo/transfer.py", line 210, in <module>
main()
File "demo/transfer.py", line 178, in main
engine = ImprintingEngine(args.extractor)
File "/usr/lib/python3/dist-packages/edgetpu/swig/edgetpu_cpp_wrapper.py", line 465, in init
this = _edgetpu_cpp_wrapper.new_ImprintingEngine(model_path)
RuntimeError: Embedding extractor's output tensor should be [1, 1, 1, x]

2、
---------------- Start training -----------------
Traceback (most recent call last):
File "demo/classification_transfer_learning.py", line 210, in <module>
main()
File "demo/classification_transfer_learning.py", line 179, in main
labels_map = engine.TrainAll(train_input)
File "/usr/lib/python3/dist-packages/edgetpu/learn/imprinting/engine.py", line 35, in TrainAll
ret[self.Train(tensors)] = category
File "/usr/lib/python3/dist-packages/edgetpu/swig/edgetpu_cpp_wrapper.py", line 492, in Train
return _edgetpu_cpp_wrapper.ImprintingEngine_Train(self, input)
TypeError: Cannot cast array data from dtype('O') to dtype('uint8') according to the rule 'safe'

二、常用命令
1、
scp -r person mendel@192.168.100.2:~/edge_tpu_python_scripts/edgetpu/test_data