当前位置:网站首页>解决glob()返回文件排序不一致问题&onnx本地按照安装方法
解决glob()返回文件排序不一致问题&onnx本地按照安装方法
2022-08-02 03:21:00 【& Tom】
0.问题背景
将服务器RTX2080Ti上的数据迁移至自己电脑RTX3060上,一样的代码,一个是Ubuntu20.04,一个是18.04,各种软件包版本均相同。
问题:训练时损失基本是常数,变化不大,计算出来的mIoU是一个常数,没有变化,上次遇到这个问题是由于标签的问题,这次已经处理好了又出现了问题,会是哪里的问题呢?
1.寻找解决办法
半夜在Debug的时候,一步一步Debug,发现在加载数据集的时候,用到了以下代码:
self.train_img_paths = glob(self.root + r"/image/training/*")
self.train_mask_paths = glob(self.root + r"/annotation/training/*")
self.val_img_paths = glob(self.root + r"/images/test/*")
self.val_mask_paths = glob(self.root + r"/annotations/test/*")在服务器上使用,可以正常使用,在本地电脑上却出现以下问题:

可以看出,img路径和mask路径出现不对应,这也不可能正确学习到东西阿!
2.解决方法
可以看出,出现以上问题的原因基本就是因为图片和标签不对应而造成的,解决方法就是对glob到的文件按照一定的规律排序,使图像和标签对应即可。加入sorted()排序后的文件顺序如图,这下标签和图像基本对应起来了。调试以下,也一切正常了。
self.train_img_paths = sorted(glob(self.root + r"/image/training/*"))
self.train_mask_paths = sorted(glob(self.root + r"/annotation/training/*"))
self.val_img_paths = sorted(glob(self.root + r"/images/test/*"))
self.val_mask_paths = sorted(glob(self.root + r"/annotations/test/*"))
顺序正常了,对应起来,才能正常学习呀!又浪费我好几天时间。。。
----------------------------------------------------------------------------------------------------------------------------
**onnx安装包地址
https://anaconda.org/conda-forge/onnx/files
按装方法,下载.tar文件后,进入环境,使用安装命令:
conda install --use-local 你的压缩文件的名字如: conda install --use-local onnxruntime-1.11.1-py38hf502cf8_1.tar.bz2
等待安装完毕即可。
测试:
进入环境:
1.
python
import onnx没有提示报错即可。
边栏推荐
猜你喜欢
随机推荐
mysql中如何查看表是否被锁
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boo
OD-Model [4]: SSD
构造方法、方法重载、全局变量与局部变量
线性代数学习笔记2-2:向量空间、子空间、最大无关组、基、秩与空间维数
Double Strings (don't always forget substr)
磷脂-聚乙二醇-巯基,DSPE-PEG-Thiol,DSPE-PEG-SH,MW:5000
错误:with open(txt_path,‘r‘) as f: FileNotFoundError: [Errno 2] No such file or directory:
parser = argparse.ArgumentParser() parsing
【面经】米哈游数据开发一面二面面经
2022ACM夏季集训周报(五)
「PHP基础知识」PHP中对象的使用
Day34 LeetCode
Redis的集群模式
[详解C语言]一文带你玩转C语言小游戏---三子棋
Knowledge Engineering Assignment 2: Introduction to Knowledge Engineering Related Fields
我的两周年创作纪念日
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boo
Chapter 10 Clustering
2022年比若依更香的开源项目









