Abstract: In order to improve the communication capacity and the users’ communication quality of unmanned aerial vehicles (UAV) coverage area with millimeter-wave beam, combined with uniform planar array antenna, a beam optimization method based on the quantization of target area is proposed in this paper. The method quantizes the target region firstly, and obtains the ideal hybrid precoding vector according to the ideal beam gain. Secondly, the mean square error between the actual and ideal precoding vector is minimized to obtain the optimal beam, and further an improved orthogonal matching pursuit（OMP）algorithm is used to fit the beamforming vector suitable for the actual system. Simulation proves that the proposed beam design method is superior to the traditional method in convergence speed and convergence precision, at the same time, this method can obtain better beam shape and gain for improving the communication quality of users in the coverage area of the UAV.