Wastewater irrigation is a very important resource for heavy metal pollution in soil and then accumulation in vegetable crops. In this study, a polyethylenimine (PEI)-grafted gelatin sponge was prepared to effectively adsorb heavy metals during wastewater irrigation. Based on the strong water adsorption ability, wastewater remained in the PEI-grafted gelatin sponge for a sufficient time for the heavy metals to interact with the sorbents. The binding capacities of Pb(II) ions and Cd(II) ions on the PEI-grafted gelatin sponge were 66 mg g(-1) and 65 mg g(-1), which were much more than those on the gelatin sponge (9.75 mg g(-1) and 9.35 mg g(-1)). Subsequently, the PEI-grafted gelatin sponge was spread on the surface of soil planted with garlic and then sprayed with synthetic wastewater. The concentrations of cadmium and lead in the garlic leaves were 1.59 mg kg(-1) and 5.69 mg kg(-1), respectively, which were much lower than those (15.78 mg kg(-1) and 27.98 mg kg(-1)) without the gelatin sponge, and the removal efficiencies were 89.9% and 79.7%. The PEI-grafting gelatin sponge could effectively remove heavy metals during wastewater irrigation, which improved the soil environment and reduced human exposure to heavy metals.