Commit Graph

7 Commits

Author SHA1 Message Date
Raphael Maenle
4af63a8520 adds image visualization 2022-03-17 11:49:00 +01:00
Raphael Maenle
d7ab431fd6 publishes bounding boxes
- bounding boxes found in image over 0.1
  score are published as a Detection2D
  array
- post-processing moved to seperate thread
2022-03-14 17:42:34 +01:00
Raphael Maenle
810f412821 ros wrapper for hailo inference
- hailo infer node subscribes to an
  image topic
- preprocess image via PIL to the
  desired image parameters (hard-
  coded currently for yolov5m)
- processes inference on hailo chip
- post process and prints bounding boxes
  into terminal
2022-03-14 12:37:05 +01:00
Raphael Maenle
43d02593f2 hailo async processing, data, hef
- hef files for darias yolov5, yolov5m
  and yolov5s added to repo
- minimal data example with image and
  daria / coco labels
- inference now allows an async call,
  which keeps the connection to the hailo
  device open, which allows for a higher
  fps rate (as shown in the test example)
2022-03-02 16:43:07 +01:00
raphael
be89e739b4 refactors into classes
- ImageMeta struct saves image metadata
- DataHandler takes care of loading dataset
  and parsing the label information
- YoloProcessing takes care of image preprocessing
  and yolo postprocessing
- HailoHandler connects to Hailo device, pushes
  the desired network hef file and runs the dataset
  on the hailo chip
2022-03-02 11:44:50 +01:00
raphael
c64b8a27dd adds inference tutorial setup for custom yolov5
- postprocessing for yolov5 included
- loading hef file of custom yolov5 onto hailo chip
- running inference
- saving image and printing fps for inference
  and postprocessing
2022-02-24 16:49:23 +01:00
Raphael Maenle
4baf090251 Initial commit 2022-02-24 15:12:33 +00:00