MNER
Requirements(Ubuntu 18.04.5 LTS, Nvidia Titan Xp)
python>=3.6
torch>=1.7.1
pytorch-crf>=0.7.2
gensim==3.8.3
Pre-trained Models
Download the pre-trained ResNet-152 via this link (https://download.pytorch.org/models/resnet152-b121ed2d.pth) Download the pre-trained Mask-RCNN via this link (https://github.com/matterport/Mask_RCNN/releases/download/v2.0/mask_rcnn_coco.h5)
Datasets
You can download the two twitter datasets from here
Training for MNER
This is the training code of tuning parameters on the dev set, and testing on the test set:
sh run_mtmner_crf.sh