如果您想要获得特定图像的解码器坐标,可以在训练过程中使用固定的随机种子来确保数据加载的一致性。这样,每次训练时使用相同的种子,相同的图像将始终被分配给相同的随机块,并且您可以在解码器中获取与该图像对应的坐标。另外,您也可以在训练集中添加图像的元数据,例如图像名称或ID,以便在解码器中进行匹配。

DETR加载数据的时候会随机选取这样的话我就没有办法准确的得到decoder中所得到的坐标是针对具体哪一张图片的坐标了

原文地址: https://www.cveoy.top/t/topic/cUxu 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录