• 搭建环境
• 准备数据集
• • 官方数据集
  • • PASCAL_VOC 07+12 and Clipart
• 预训练模型
• 训练模型
• 测试模型
• 训练结果

搭建环境

克隆项目HTCN

git clone git://github.com/xiangxiangtao/HTCN.git

主要环境

• python=3.6.9
• pytorch=1.0
• cuda=10.2

其他环境参考fasterrcnn所需环境

编译

cd lib

python setup.py build develop

准备数据集

官方数据集

PASCAL_VOC 07+12 and Clipart

准备PASCAL_VOC 07+12
准备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

报错

ModuleNotFoundError: No module named ‘model.utils.cython_bbox’

测试模型

net根据训练时取的主干网络进行选择

python test_net_HTCN.py --dataset source_dataset --dataset_t target_dataset --net vgg16 --load_name path_to_model

训练结果