diff --git a/INSTALL.md b/INSTALL.md index 32e4d0010c220c07ae7a04f0395e1eac38047bb2..46001250a323b61067a82718794fb0c1f4e69743 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -6,7 +6,7 @@ - Python 3.4+ - PyTorch 1.0 - Cython -- [mmcv](https://github.com/open-mmlab/mmcv) >= 0.2.2 +- [mmcv](https://github.com/open-mmlab/mmcv) ### Install mmdetection diff --git a/README.md b/README.md index 7861d5bc66c463aa591bc91b99dc3a11fb0f4d55..aeb674d1d4741b1d2e4683b24e6aaf3e6b04675b 100644 --- a/README.md +++ b/README.md @@ -39,6 +39,11 @@ This project is released under the [Apache 2.0 license](LICENSE). ## Updates +v0.6.0 (14/04/2019) +- Up to 30% speedup compared to the model zoo. +- Support both PyTorch stable and nightly version. +- Replace NMS and SigmoidFocalLoss with Pytorch CUDA extensions. + v0.6rc0(06/02/2019) - Migrate to PyTorch 1.0. diff --git a/setup.py b/setup.py index 724adf38b7c821d62595f30892d48742ccda2e73..b79eb4d26cad812cdabff4cc3a53e200ddc802a3 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,7 @@ def readme(): MAJOR = 0 MINOR = 6 -PATCH = 'rc0' +PATCH = 0 SUFFIX = '' SHORT_VERSION = '{}.{}.{}{}'.format(MAJOR, MINOR, PATCH, SUFFIX) @@ -106,7 +106,7 @@ if __name__ == '__main__': setup_requires=['pytest-runner'], tests_require=['pytest'], install_requires=[ - 'mmcv', 'numpy', 'matplotlib', 'six', 'terminaltables', + 'mmcv>=0.2.6', 'numpy', 'matplotlib', 'six', 'terminaltables', 'pycocotools' ], zip_safe=False)