diff --git a/cmake/package.cmake b/cmake/package.cmake
index d6d5e3a73e..c3d9bce0f2 100644
--- a/cmake/package.cmake
+++ b/cmake/package.cmake
@@ -317,6 +317,18 @@ install(
     COMPONENT mindspore
 )
 
+## Public header files for minddata
+install(
+    FILES ${CMAKE_SOURCE_DIR}/mindspore/ccsrc/minddata/dataset/include/constants.h
+          ${CMAKE_SOURCE_DIR}/mindspore/ccsrc/minddata/dataset/include/status.h
+          ${CMAKE_SOURCE_DIR}/mindspore/ccsrc/minddata/dataset/include/transforms.h
+          ${CMAKE_SOURCE_DIR}/mindspore/ccsrc/minddata/dataset/include/vision.h
+          ${CMAKE_SOURCE_DIR}/mindspore/ccsrc/minddata/dataset/include/vision_lite.h
+          ${CMAKE_SOURCE_DIR}/mindspore/ccsrc/minddata/dataset/include/minddata_eager.h
+    DESTINATION ${INSTALL_BASE_DIR}/include/minddata/dataset/include
+    COMPONENT mindspore
+)
+
 ## config files
 install(
     FILES ${CMAKE_SOURCE_DIR}/config/op_info.config
diff --git a/mindspore/ccsrc/minddata/dataset/include/vision_lite.h b/mindspore/ccsrc/minddata/dataset/include/vision_lite.h
index e85f0cf30d..9043d043f0 100644
--- a/mindspore/ccsrc/minddata/dataset/include/vision_lite.h
+++ b/mindspore/ccsrc/minddata/dataset/include/vision_lite.h
@@ -22,9 +22,9 @@
 #include <string>
 #include <utility>
 #include <vector>
-#include "minddata/dataset/core/constants.h"
+#include "minddata/dataset/include/constants.h"
 #include "minddata/dataset/include/transforms.h"
-#include "minddata/dataset/util/status.h"
+#include "minddata/dataset/include/status.h"
 
 #ifdef ENABLE_ANDROID
 #include "minddata/dataset/kernels/image/rotate_op.h"