diff --git a/yolov5/yolov5_seg_trt.py b/yolov5/yolov5_seg_trt.py
index 7942815..bdc0465 100644
--- a/yolov5/yolov5_seg_trt.py
+++ b/yolov5/yolov5_seg_trt.py
@@ -13,7 +13,6 @@ import numpy as np
 import pycuda.autoinit
 import pycuda.driver as cuda
 import tensorrt as trt
-from scipy import ndimage
 
 CONF_THRESH = 0.5
 IOU_THRESHOLD = 0.4