Skip to content

README.md

Latest commit

 

History

History
54 lines (47 loc) · 1.08 KB

README.md

File metadata and controls

54 lines (47 loc) · 1.08 KB

split_inference

Configuration

Application configuration is in the config.yaml file:

name: YOLO
server:
  cut-layer: a #or b, c, d
  clients:
    - 1
    - 1
  model: yolo11n
  batch-frame: 1
rabbit:
  address: 127.0.0.1
  username: admin
  password: admin
  virtual-host: /

data: video.mp4
log-path: .
control-count: 10
debug-mode: False
compress:
  enable: False
  num_bit: 8
cal_map:
  enable: False
  create_label: False

This configuration is use for server.

How to Run

Alter your configuration, you need to run the server to listen and control the request from clients.

Server

python server.py

Client

Now, when server is ready, run clients simultaneously with total number of client that you defined.

Layer 1

python client.py --layer_id 1

Where:

  • --layer_id is the ID index of client's layer, start from 1.

If you want to use a specific device configuration for the training process, declare it with the --device argument when running the command line:

python client.py --layer_id 1 --device cpu