RaspberryPi 4BにTensorflow2をぶち込む

実際のところ

$ uname -a
Linux XXX 5.15.84-v8+ #1613 SMP PREEMPT Thu Jan 5 12:03:08 GMT 2023 aarch64 GNU/Linux

今回は実行するフレームワークがRasPiとお世辞にも強いとは言えない環境なのでtiny版でビルドします

sudo pip install matplotlib
sudo pip install tensorflow
cd
git clone https://github.com/neuralassembly/tensorflow2-yolo-v3
cd tensorflow2-yolo-v3/

wget https://pjreddie.com/media/files/yolov3-tiny.weights -O data/yolov3-tiny.weights
python3 convert.py —weights ./data/yolov3-tiny.weights —output ./checkpoints/yolov3-tiny.tf —tiny

使うには、こう

python3 detect.py —weights ./checkpoints/yolov3-tiny.tf —tiny —image ./data/street.jpg

USB カメラ

USBカメラが繋がっていれば、メモリ8Gでも多少カクつくものの動きます

$ python3 detect.py —weights ./checkpoints/yolov3-tiny.tf —tiny —webcam

参考もと

カラー図解 Raspberry Piではじめる機械学習 補足情報: Raspberry Pi で YOLO v3-Tiny / YOLO v3 による物体検出を試してみよう
YOLOv3 - 自作データセットで学習する方法について - pystyle