DS-DETR: A Model for Tomato Leaf Disease Segmentation and Damage Evaluation

Jianshuang Wu, Changji Wen, Hongrui Chen, Zhenyu Ma, Tian Zhang, Hengqiang Su, Ce Yang

Research output: Contribution to journalArticlepeer-review

4 Scopus citations


Early blight and late blight are important factors restricting tomato yield. However, it is still a challenge to accurately and objectively detect and segment crop diseases in order to evaluate disease damage. In this paper, the Disease Segmentation Detection Transformer (DS-DETR) is proposed to segment leaf disease spots efficiently based on several improvements to DETR. Additionally, a damage assessment is carried out by the area ratio of the segmented leaves to the disease spots. First, an unsupervised pre-training method was introduced into DETR with the Plant Disease Classification Dataset (PDCD) to solve the problem of the long training epochs and slow convergence speed of DETR. This method can train the Transformer structures in advance to obtain leaf disease features. Loading the pre-training model weight in DS-DETR can speed up the convergence speed of the model. Then, Spatially Modulated Co-Attention (SMCA) was used to assign Gaussian-like spatial weights to the query box of DS-DETR. The different positions in the image are trained using the query boxes with different weights to improve the accuracy of the model. Finally, an improved relative position code was added to the Transformer structure of DS-DETR. Relative position coding promotes the capture of the sequence order of input tokens by the Transformer. The spatial location feature is strengthened by establishing the location relationship between different instances. Based on these improvements, the DS-DETR model was tested on the Tomato leaf Disease Segmentation Dataset (TDSD) constructed by us. The experimental results show that the DS-DETR proposed by us achieved 0.6823 for APmask, which improved by 12.87%, 8.25%, 3.67%, 1.95%, 10.27%, and 9.52% compared with the state-of-the-art: Mask RCNN, BlendMask, CondInst, SOLOv2, ISTR, and DETR, respectively. In addition, the disease grading accuracy reached 0.9640 according to the segmentation results given by our proposed model.

Original languageEnglish (US)
Article number2023
Issue number9
StatePublished - Sep 2022

Bibliographical note

Funding Information:
This research was funded by the National Natural Science Foundation of China (Key Program) (No. U19A2061), the Industrial Technology and Development Project of Development and Reform Commission of Jilin Province (No. 2021C044-8), the Natural Science Foundation of Jilin Province of China (No. 20180101041JC), the Social Sciences project of Jilin Provincial Education Department (No. JJKH20220376SK), and Science and technology research project of Education Department of Jilin Province (No. JJKH20190924KJ).

Publisher Copyright:
© 2022 by the authors.


  • deep learning
  • detection transformer
  • disease damage evaluation
  • instance segmentation
  • tomato disease


Dive into the research topics of 'DS-DETR: A Model for Tomato Leaf Disease Segmentation and Damage Evaluation'. Together they form a unique fingerprint.

Cite this