Augmented reality head-mounted displays (AR-HMDs) based on diffractive waveguides have been a challenging and rewarding research topic focusing on near-eye displays. The size of the exit pupil and uniformity of the image illuminance are two important factors that affect the display performance of the diffractive waveguide. In this paper, a novel method for optimizing high uniformity of two-dimensional (2D) diffractive waveguide is proposed. A straight-line 2D surface relief grating (SRG) waveguide with divided grating regions is designed. An illuminance uniformity evaluation model of the energy propagation process is established, and non-sequential ray tracing is utilized to optimize the diffraction efficiency of multi-regions grating to achieve illuminance uniformity distribution. Then, the uniformity distribution of the diffraction efficiency in different fields of view (FOVs) is realized by combining the particle swarm optimization (PSO) algorithm and rigorous couple wave analysis (RCWA) to optimize the grating structural parameters, which further ensures the uniformity of the exit pupil illuminance and angular illuminance. The waveguide with exit pupil expansion (EPE) has exit pupil size of 16 mm × 14 mm at an eye relief (ERF) of 20 mm, exit pupil illuminance uniformity of 91%, and angular uniformity illuminance of 64%.