交流网
这是官方的工具
, , Lei Fei,
Accepted by IEEE ICIP 2019
实验结果
我们在SUN-RGBD和NYUDv2中评估ACNet的mIoU
太阳RGBD
纽约市第二版
密欧
48.1%
48.3%
如何使用
这段代码是NYUDv2工具。
要求
Python 3
Pytorch 0.4.1
TensorboardX
Tensorboard
训练模式
python ACNet_train_V1_nyuv2.py --cuda -b 4
评估模式
NYUDv2的预训练模型可从。
带有.npy格式的预处理NYUDv2数据集可在,密码:k6rv
python ACNet_eval_nyuv2.py --cuda --last-ckpt '.\model\ckpt_epoch_1195.00.pth'
注意:该代码部分基于RedNet(
1