#9 fixed YOLOX inference error for non-square shape

DefTruth · Jul 31, 2021 · d243458 · d243458
1 parent c7eb55c
commit d243458
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/lite/lite.h b/lite/lite.h
@@ -454,7 +454,7 @@ namespace lite
         typedef _ONNXMobileNetV268 MobileNetV268;
         typedef _ONNXMobileNetV2SE68 MobileNetV2SE68;
         typedef _ONNXFaceLandmark1000 FaceLandmark1000;
-        typedef _ONNXRetinaFace _RetinaFace;
+        typedef _ONNXRetinaFace RetinaFace;
 
         namespace detect
         {

diff --git a/ort/core/ort_core.h b/ort/core/ort_core.h
@@ -68,6 +68,17 @@ namespace ortcv
   class LITE_EXPORTS RetinaFace;                 // [54] * reference: https://github.com/biubug6/Pytorch_Retinaface
   class LITE_EXPORTS FaceBoxes;                  // [55] reference: https://github.com/zisianw/FaceBoxes.PyTorch
   class LITE_EXPORTS YoloX;                      // [56] * reference: https://github.com/Megvii-BaseDetection/YOLOX
+  class LITE_EXPORTS Pyramidbox;                 // [57] reference: https://github.com/yxlijun/Pyramidbox.pytorch
+  class LITE_EXPORTS TinyFaceDet;                // [58] reference: https://github.com/varunagrawal/tiny-faces-pytorch
+  class LITE_EXPORTS S3FD;                       // [59] reference: https://github.com/yxlijun/S3FD.pytorch
+  class LITE_EXPORTS DSFD;                       // [60] reference: https://github.com/yxlijun/DSFD.pytorch
+  class LITE_EXPORTS TencentDSFD;                // [61] reference: https://github.com/Tencent/FaceDetection-DSFD
+  class LITE_EXPORTS BlazeFace;                  // [62] reference: https://github.com/hollance/BlazeFace-PyTorch
+  class LITE_EXPORTS GoinPyramidBox;             // [63] reference: https://github.com/Goingqs/PyramidBox
+  class LITE_EXPORTS CenterFace;                 // [64] reference: https://github.com/chenjun2hao/CenterFace.pytorch
+  class LITE_EXPORTS FaceMaskDetection;          // [65] reference: https://github.com/AIZOOTech/FaceMaskDetection
+  class LITE_EXPORTS DBFace;                     // [66] reference: https://github.com/dlunion/DBFace
+
 }
 
 namespace ortnlp