Skip to content

Darknet binding written in Go

Compare
Choose a tag to compare
@LdDl LdDl released this 26 Feb 05:24
· 64 commits to master since this release
630ded7

Since this repository https://github.com/gyonluks/go-darknet is no longer maintained I decided to move on and make little different bindings for Darknet.
This bindings aren't for official implementation but for AlexeyAB's fork.

Added features:

  1. It is possible to load image.Image into []float32 slice and inference it to CGO. https://github.com/LdDl/go-darknet/blob/master/image.go#L62
  2. Has GPU support (CUDA).
  3. Has download_data.sh scpirt to load YOLOv3 weights/cfg/classnames and sample image.

Future works:

  1. Add classic function for loading image from file system.
  2. Save predictions
  3. gRPC server example
  4. Updates according to AlexeyAB's fork