diff --git a/main.sh b/main.sh
index 3176831..b41de18 100644
--- a/main.sh
+++ b/main.sh
@@ -66,6 +66,11 @@ build(){
   # ────────────────────────────────────────────
   EXTRA_IMAGE_NAME=""
 
+  # ────────────────────────────────────────────
+  #  根分区大小（单位：MB），留空则使用 imagebuilder 默认值
+  # ────────────────────────────────────────────
+  ROOTFS_PARTSIZE="${2:-}" 
+
   # =============================================================
   #  插件包列表
   #  - 前面加 - 表示从固件中移除该包，例如 -dnsmasq
@@ -162,7 +167,8 @@ build(){
       PACKAGES="$PACKAGES" \
       FILES="$FILES_DIR" \
       BIN_DIR="$OUTPUT_DIR" \
-      EXTRA_IMAGE_NAME="$EXTRA_IMAGE_NAME"
+      EXTRA_IMAGE_NAME="$EXTRA_IMAGE_NAME" \
+      ${ROOTFS_PARTSIZE:+ROOTFS_PARTSIZE="$ROOTFS_PARTSIZE"} # ROOTFS_PARTSIZE 参数不为空才添加
 
   # 构建结果判断
   if [ $? -eq 0 ]; then
@@ -217,4 +223,4 @@ install_dependencies
 builder_dir="$(download_imagebuilder "$version")"
 
 # 构建固件
-build $builder_dir
+build $builder_dir 500   # 500 表示根分区大小为 500MB