We provide our dataset and PyTorch implementation for relation network benchmark. Details are in our paper.
- Linux or macOS
- Python 3
- CPU or NVIDIA GPU + CUDA CuDNN
- PyTorch 0.4+
- Google drive: download here
- Online Preview
First, download pretrained model here.
python autolabel.py -sd imgs/example/support -td imgs/example/query
- Set option
-sd
to the support directory and the script will input them as support set. - Set option
-td
to the path of your query images. - Results will be saved under
./results
View our testing results here
- Label 5 support images following the format in
imgs/example/support/
. - Set your support and query path accordingly.
Coming soon