dkurt / hair_segmentation Goto Github PK

This repository contains a simple Python script which runs Mediapipe's hair segmentation model using OpenCV on RISC-V CPU.

1. Download a compiler: Xuantie-900-gcc-linux-5.10.4-glibc-x86_64-V2.6.1-20220906.tar.gz and put to opencv_riscv64 folder without unpacking.

2. Build OpenCV:

   docker build -t opencv opencv_riscv64
   docker run -v $(pwd):/mnt opencv sh -c "cp *.tar.gz /mnt"

3. Download deep learning model

4. Start a telegram bot:

   python bot.py --token=xxx

HW: Sipeed Lichee RV Dock (Allwinner D1 aka XuanTie C906 CPU)

OS: 20211230_LicheeRV_debian_d1_hdmi_8723ds