/IIE4MNER

Primary LanguagePythonMIT LicenseMIT

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