yongtaoge / point-teaching

Point-teaching: Weakly semi-supervised Object Detection with Point Annotations (AAAI 2023)

Install

pip install detectron2
pip install -e . -v

Train

# 10% coco subset
sh scripts/train_frcnn_10.0_ins_mil_point_match.sh

Test

sh scripts/test_frcnn.sh

🎓 Citation

@inproceedings{ge2023point,
  title={Point-teaching: Weakly semi-supervised object detection with point annotations},
  author={Ge, Yongtao and Zhou, Qiang and Wang, Xinlong and Shen, Chunhua and Wang, Zhibin and Li, Hao},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  volume={37},
  number={1},
  pages={667--675},
  year={2023}
}

point-teaching's Issues

run code?

感谢您的贡献
请问可以更新下代码运行的详细配置和运行过程吗。

期待您的回复

FileNotFoundError: [Errno 2] No such file or directory: 'datasets/coco/annotations/instances_train2017_n2_v1.json'

What‘s the organized format of the dataset in the code, and is there a need for an additional annotation format?

ImportError

Codes in ms_deform_attn_func.py raise this error: "ImportError: cannot import name '_C' from 'pteacher'".Could you please check it!

README不完整

Hi, when will you be able to update the environment configuration and train/test process of the code in the README？ I tried to run pip install -e . but I got an error; moreover, I see that the code involves a call to ubteacher library, which is not in the native detectron2 environment.

Detectron2 版本

what‘s the verson of detectron2 in your code environment？as well as pytorch.

source code

May i ask when will the code be made public？