这是数据集,可以以一定方式下载
下载后:
然后跑程序会出现很多问题,删除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