An implementation of Non-local block intruduced in Tensorflow. Link to the original paper:Non-local Neural Networks
This repository includes the code(training and testing) for DRN and Non-local. The code is for semantic segmentation.
python 2.7.x
tensorflow >= 1.4.0
-
Download VOC2012 dataset.
-
Prepare the tfrecord data to train the model. Please see my another repository Convert_To_TFRecord.
-
Train the model
#the parameter of --mode in main.py need to be "train"
python main.py
- Test the model
#the parameter of --mode in main.py need to be "test"
python main.py
Method | Non-local | mIoU |
---|---|---|
DRN | ✗ | 7.31 |
DRN | ✓ | 0.094 |
The result is so terrible, non-local has a bad influence, I have no idea about it, what's wrong with this? Is it because there is no pretrainig?
- COCO
- Cityscapes
- ADE