Travel restrictions have become an important epidemic preventive measure, but there are few relevant quantitative studies. In this paper, travel proportion is introduced into a four-compartment model to quantify the spread of COVID-19 in Wuhan. It is found that decreasing the travel proportion can reduce the peak of infections and delay the peak time. When the travel proportion is less than 35%, transmission can be prevented. This method provides reference for other places.