【文件属性】:
文件名称:MIAL:用于对象检测的多实例主动学习的代码,CVPR 2021
文件大小:12.63MB
文件格式:ZIP
更新时间:2021-03-03 15:44:36
mil object-detection active-learning cvpr multiple-instance-learning
语言:|
介绍
这是(CVPR 2021)的代码。
安装
推荐使用Linux平台(我们是Ubuntu 18.04 LTS)和 ,因为它们可以方便,高效地安装和管理环境和软件包。
建议使用TITAN V GPU和带有 ,因为它们可以加快模型训练的速度。
安装anaconda3之后,您可以创建一个conda环境,如下所示:
conda create -n mial python=3.7 -y
conda activate mial
有关环境安装,请参阅和其 。
mmcv软件包中的修改
为了两个dataloaders训练(即,标记集的DataLoader,并在文件中提到的未标记集的DataLoader)在同一时间,你将需要修改epoch_based_runner.py在mmcv包。
考虑到这将影响使用该环境的所有代码,因此我们建议您为MIAL建立一个单独的环境(即,上面创建的mial环境