【复现2】CADTransformer: Panoptic Symbol Spotting Transformer for CAD Drawings

请添加图片描述
这是数据集,可以以一定方式下载
下载后:
请添加图片描述
然后跑程序会出现很多问题,删除svg2png.py文件中,93行的一部分,然后:
请添加图片描述
这里改了99行和100行代码
然后再跑代码
又出现两个问题:
请添加图片描述

请添加图片描述
cp的问题我记得改代码sudo好像无法实现,通过授权指定文件夹所有权来实现

sudo chown -R myz ssd1

myz是我的计算机名字

然后还有signal报错,是在_utils_dataset.py文件中加入头文件

import signal

不出所料,再次报错:
请添加图片描述
这里我尝试用sudo apt autoremove cairosvg ,结果是不行的,“-o”参数不可用
然后用pip下载:

pip3 install cairosvg

再次跑代码,稍等片刻(25min)
这个时间可以缩短修改_utils_dataset.py文件的350行改成

time.sleep(0.001)

跑完报错:
请添加图片描述
解决方法
其中有两种方法,我用了第二种,安装库xml

pip install lxml

然后再次运行(20min,没有快多少):
运行成功:
请添加图片描述
查看文件夹
请添加图片描述然后接下来继续跑代码:

generate npy format data:

python preprocess/preprocess_svg.py -i /ssd1/zhiwen/datasets/svg_processed/svg/train  -o /ssd1/zhiwen/datasets/svg_processed/npy/train   --thread_num  48
python preprocess/preprocess_svg.py -i /ssd1/zhiwen/datasets/svg_processed/svg/test  -o /ssd1/zhiwen/datasets/svg_processed/npy/test   --thread_num  48
python preprocess/preprocess_svg.py -i /ssd1/zhiwen/datasets/svg_processed/svg/val  -o /ssd1/zhiwen/datasets/svg_processed/npy/val   --thread_num  48

成功:
请添加图片描述
然后接着做,先对照一下README:
请添加图片描述
最后发现跑出来只有npy文件夹

请添加图片描述
找一下原因请添加图片描述

我在代码中加入了输出查看了一下,感觉作者的意思是要吧 /ssd1/zhiwen/datasets/svg_raw 中的两个文件夹放到 /ssd1/zhiwen/datasets/svg_processed中
尝试一下,然后再运行

python preprocess/preprocess_svg.py -i /ssd1/zhiwen/datasets/svg_processed/svg/train  -o /ssd1/zhiwen/datasets/svg_processed/npy/train   --thread_num  48
python preprocess/preprocess_svg.py -i /ssd1/zhiwen/datasets/svg_processed/svg/test  -o /ssd1/zhiwen/datasets/svg_processed/npy/test   --thread_num  48
python preprocess/preprocess_svg.py -i /ssd1/zhiwen/datasets/svg_processed/svg/val  -o /ssd1/zhiwen/datasets/svg_processed/npy/val   --thread_num  48

再等一段时间(2h以上)空间不是很够腾一下
fuck

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值