搭建环境
克隆项目HTCN
git clone git://github.com/xiangxiangtao/HTCN.git
主要环境
- python=3.6.9
- pytorch=1.0.1
- cuda=10.0
其他环境参考fasterrcnn所需环境
编译
cd lib
python setup.py build develop
准备数据集
下载数据集
PASCAL_VOC 07+12 and Clipart
准备PASCAL_VOC 07+12
准备Clipart
修改数据集配置信息
添加lib/datasets/
将数据集对应的脚本添加到lib/datasets/中
- 对voc数据集，lib/datasets/中已经有了对应的脚本pascal_voc.py，不需要再添加；
- 对clipart数据集，lib/datasets/中已经有了对应的脚本clipart.py，不需要再添加，不过需要根据实际需要修改clipart.py脚本中的路径，否则代码中的默认路径和clipart默认的路径组织方式不一样；
修改lib/datasets/config_dataset.py
- 修改__D.PASCAL和__D.CLIPART为实际的数据集所在路径；
修改lib/model/utils/parser_func.py
- 对voc和clipart两个数据集，–dataset的值改为’voc_0712’，–dataset_t的值改为’clipart’，由于该脚本中set_dataset_args函数里已经有了对应的分支，不需要再进行修改；
修改lib/datasets/factory.py
根据前面的parser_func.py脚本中的args.imdb_name内容进行修改
- args.dataset为"voc_0712"时，使用默认的代码即可；
- args.dataset_t为"clipart"时，使用默认的代码即可；
预训练模型
下载预训练模型
点击HTCN中的链接，下载VGG16和ResNet101的预训练模型；
将预训练模型放到指定路径下，指定路径可以在lib/model/utils/config.py中进行修改；
训练模型
net可以选择vgg16/resnet101
python trainval_net_HTCN.py --dataset source_dataset --dataset_t target_dataset --net vgg16
报错1：ModuleNotFoundError: No module named ‘model.utils.cython_bbox’
解决办法：
打开setup.py
gedit lib/setup.py
将里面的
ext_modules = [
extension(
"model._C",
sources,
include_dirs=include_dirs,
define_macros=define_macros,
extra_compile_args=extra_compile_args,
)
]
替换成
ext_modules = [ extension( "model._C", sources, include_dirs=include_dirs, define_macros=define_macros, extra_compile_args=extra_compile_args, ), extension( "model.utils.cython_bbox", ["model/utils/bbox.pyx"], extra_compile_args={'cxx':[],'gcc': ["-Wno-cpp", "-Wno-unused-function"]}, include_dirs=[numpy_include] ) ]
报错2：NameError: name ‘numpy_include’ is not defined
解决办法
打开setup.py
gedit lib/setup.py
在刚才的ext_modules前面添加下面几行内容
import numpy as np
try:
numpy_include = np.get_include()
except AttributeError:
numpy_include = np.get_numpy_include()
报错3：AttributeError: ‘Namespace’ object has no attribute ‘set_cfgs’
解决办法：
数据集没配置好，参考前面的内容进行配置；
报错4：RuntimeError: set_sizes_contiguous is not allowed on a Tensor created from .data or .detach().
解决办法：
参考报错提示进行修改，去掉data。
测试模型
net根据训练时取的主干网络进行选择
python test_net_HTCN.py --dataset source_dataset --dataset_t target_dataset --net vgg16 --load_name path_to_model
训练结果
[1]：Pascal VOC数据集使用方法
[2]：HTCN
[3]：ubuntu安装pytorch及使用过程的报错记录