很多数据集中的类别过多,而且有很多不是我们需要的,可以通过下面简短的python代码来删掉不需要的类别。
以Kitti数据集为例:
import xml.etree.cElementTree as ET
import os
path_root = ['/Applications/小仙女/Annotations'] #自己的xml路径
CLASSES = [ "car","truck","van",
很多数据集中的类别过多,而且有很多不是我们需要的,可以通过下面简短的python代码来删掉不需要的类别。
以Kitti数据集为例:
import xml.etree.cElementTree as ET
import os
path_root = ['/Applications/小仙女/Annotations'] #自己的xml路径
CLASSES = [ "car","truck","van",