Update model zoo

Author: yeliudev

README.md

@@ -194,13 +194,13 @@ We provide multiple HICO-DET pre-trained models here. All the models are trained
       <a href="https://github.com/yeliudev/ConsNet/blob/main/configs/consnet_5e_hico_det.py">ConsNet</a>
     </td>
     <td align="center">—</td>
-    <td align="center">—</td>
-    <td align="center">—</td>
-    <td align="center">—</td>
+    <td align="center">22.15</td>
+    <td align="center">17.55</td>
+    <td align="center">23.52</td>
     <td align="center">—</td>
     <td align="center">—</td>
     <td align="center">
-      <a href="">model</a> |
+      <a href="https://catcatserver.xyz/models/consnet/consnet_5e_hico_det/consnet_5e_hico_det-6e89e1d9.pth">model</a> |
       <a href="https://catcatserver.xyz/models/consnet/consnet_5e_hico_det/metrics.json">metrics</a>
     </td>
   </tr>

consnet/api/evaluation.py

@@ -94,7 +94,7 @@ def hico_det_eval(blob,
     Returns:
         dict: Evaluation results including mean average precision (mAP) and \
             mean recall (mRec) values of multiple dataset splits under \
-            different evaluation mode.
+            different evaluation modes.
 
     Example:
         >>> results = hico_det_eval(blob, '<path-to-anno_bbox.mat>')

consnet/models/builder.py

@@ -5,7 +5,6 @@
 # -----------------------------------------------------
 
 from nncore import Registry, build_object
-from nncore.engine import load_checkpoint, set_random_seed
 from nncore.parallel import NNDataParallel, collate
 from torch.utils.data import DataLoader
 
@@ -16,6 +15,7 @@
     from mmdet.datasets import DATASETS
     from mmdet.datasets import build_dataloader as _build_dataloader
     from mmdet.datasets import build_dataset as _build_dataset
+    from nncore.engine import load_checkpoint, set_random_seed
 except ImportError:
     pass
 

docs/index.rst

@@ -24,3 +24,18 @@ ConsNet API is a part of the `ConsNet <https://github.com/yeliudev/ConsNet>`_ pr
    consnet.api.data
    consnet.api.bbox
    consnet.api.evaluation
+
+Citation
+-------------------------
+
+If you find this project useful for your research, please kindly cite our paper.
+
+.. code-block:: text
+
+   @inproceedings{liu2020consnet,
+     title={ConsNet: Learning Consistency Graph for Zero-Shot Human-Object Interaction Detection},
+     author={Liu, Ye and Yuan, Junsong and Chen, Chang Wen},
+     booktitle={Proceedings of the 28th ACM International Conference on Multimedia},
+     pages={4235--4243},
+     year={2020}
+   }

setup.py

@@ -47,11 +47,11 @@ def get_readme():
         'Topic :: Utilities',
     ],
     python_requires='>=3.8',
-    install_requires=[
-        'nncore>=0.2.2', 'scipy>=1.6', 'torch>=1.6', 'torchvision>=0.7'
-    ],
+    install_requires=['nncore>=0.2.2', 'scipy>=1.6', 'torch>=1.6'],
     extras_require={
-        'full':
-        ['allennlp>=2.2,<2.3', 'mmcv-full>=1.3,<1.4', 'mmdet>=2.11,<2.12']
+        'full': [
+            'allennlp>=2.2,<2.3', 'mmcv-full>=1.3,<1.4', 'mmdet>=2.11,<2.12',
+            'torchvision>=0.7'
+        ]
     },
     packages=find_packages(exclude=('.github', 'configs', 'docs', 'tools')))