diff --git a/mindyolo/data/dataset.py b/mindyolo/data/dataset.py
index fc01db391948309000e624dcad84c1ec6a2cc05c..43781d4df89470540f73d300caaa496d819ec713 100644
--- a/mindyolo/data/dataset.py
+++ b/mindyolo/data/dataset.py
@@ -195,7 +195,7 @@ class COCODataset:
 
             self.batch_shapes = np.ceil(np.array(shapes) * img_size / stride + pad).astype(np.int_) * stride
 
-        self.imgIds = [int(Path(im_file).stem) for im_file in self.img_files]
+        self.imgIds = [str(Path(im_file).stem) for im_file in self.img_files]
 
     def cache_labels(self, path=Path("./labels.cache.npy")):
         # Cache dataset labels, check images and read shapes