detect.h 733 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. #ifndef DETECT_H
  2. #define DETECT_H
  3. #ifdef __cplusplus
  4. extern "C" {
  5. #endif
  6. //#include "cnndecoder.h"
  7. typedef struct BBoxOut
  8. {
  9. int label;
  10. float score;
  11. float xmin;
  12. float ymin;
  13. float xmax;
  14. float ymax;
  15. }BBoxOut;
  16. #define PIX3218 0
  17. #define PIX3030 1
  18. #if PIX3218
  19. #define num_prior 1224
  20. #elif PIX3030
  21. #define num_prior 1917
  22. #endif
  23. typedef struct BBox
  24. {
  25. float xmin;
  26. float ymin;
  27. float xmax;
  28. float ymax;
  29. }BBox;
  30. typedef struct BBoxRect
  31. {
  32. float xmin;
  33. float ymin;
  34. float xmax;
  35. float ymax;
  36. int label;
  37. }BBoxRect;
  38. int ssdforward(float *location,float * confidence,float * priorbox,BBox *bboxes,BBoxOut *out);
  39. int readbintomem(float *dst,char *path);
  40. #ifdef __cplusplus
  41. }
  42. #endif
  43. #endif