From 00c85c2aabfdd54d63b7d610c4d64fc7fc97e1c6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B8=B8=E5=BA=B7?= <1745822665@qq.com>
Date: Sat, 17 Aug 2024 00:49:05 +0000
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=A6=82=E4=B8=8B=E9=94=99?=
 =?UTF-8?q?=E8=AF=AF=EF=BC=9AValueError:=20invalid=20literal=20for=20int()?=
 =?UTF-8?q?=20with=20base=2010=20=E6=94=B9=E4=BB=A3=E7=A0=81=E7=9A=84?=
 =?UTF-8?q?=E5=90=AB=E4=B9=89=E6=98=AF=E5=B0=86=E6=89=80=E6=9C=89=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E9=9B=86=E6=96=87=E4=BB=B6=E7=9A=84=E5=90=8D=E7=A7=B0?=
 =?UTF-8?q?=E4=BF=9D=E5=AD=98=E5=88=B0=E6=95=B0=E7=BB=84=E4=B8=AD=EF=BC=8C?=
 =?UTF-8?q?=E4=BD=86=E6=98=AF=E9=9C=80=E8=A6=81=E4=BF=9D=E8=AF=81=E6=89=80?=
 =?UTF-8?q?=E6=9C=89=E6=95=B0=E6=8D=AE=E9=9B=86=E7=9A=84=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E5=90=8D=E7=A7=B0=E5=BF=85=E9=A1=BB=E6=98=AF=E6=95=B0=E5=AD=97?=
 =?UTF-8?q?=E6=89=8D=E5=8F=AF=E4=BB=A5=E4=BD=BF=E7=94=A8int()=EF=BC=8C?=
 =?UTF-8?q?=E4=BD=86=E6=98=AF=E5=AE=9E=E9=99=85=E4=B8=8A=E5=A4=A7=E9=83=A8?=
 =?UTF-8?q?=E5=88=86=E6=95=B0=E6=8D=AE=E9=9B=86=E7=9A=84=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E5=90=8D=E7=A7=B0=E4=B8=8D=E6=98=AF=E6=95=B0=E5=AD=97=E3=80=82?=
 =?UTF-8?q?=E8=BF=99=E6=A0=B7=E6=94=B9=E5=8A=A8=E5=90=8E=EF=BC=8C=E7=BB=8F?=
 =?UTF-8?q?=E8=BF=87=E6=B5=8B=E8=AF=95=EF=BC=8C=E5=8F=AF=E4=BB=A5=E6=AD=A3?=
 =?UTF-8?q?=E5=B8=B8=E8=AE=AD=E7=BB=83=E5=AE=8C=E6=88=90=EF=BC=8C=E5=B9=B6?=
 =?UTF-8?q?=E4=B8=94=E4=B8=8D=E4=BC=9A=E6=8A=A5=E9=94=99=EF=BC=8C=E4=B9=9F?=
 =?UTF-8?q?=E5=85=BC=E5=AE=B9=E6=89=80=E6=9C=89=E6=A0=BC=E5=BC=8F=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E9=9B=86=E5=90=8D=E7=A7=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 常康 <1745822665@qq.com>
---
 mindyolo/data/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindyolo/data/dataset.py b/mindyolo/data/dataset.py
index fc01db3..43781d4 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
-- 
Gitee