Skip to content

Commit 5677756

Browse files
bryanyzhuWeisu Yin
andauthored
Update readme with AutoGluon (#1766)
* introduce ag * fix opencv * fix * fix * test * test * test * test * fix * test * fix six * address comments Co-authored-by: Weisu Yin <[email protected]>
1 parent 40a2165 commit 5677756

File tree

5 files changed

+11
-28
lines changed

5 files changed

+11
-28
lines changed

.github/workflows/unittest.yml

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -39,17 +39,9 @@ jobs:
3939
shell: bash -l {0}
4040
run: |
4141
conda env create -n gluon_cv_py3_test -f tests/py3_mxnet_ci.yml
42-
conda env update -n gluon_cv_py3_test -f tests/py3_mxnet_ci.yml --prune
4342
conda activate gluon_cv_py3_test
44-
conda list
45-
export CUDA_VISIBLE_DEVICES=0
46-
export KMP_DUPLICATE_LIB_OK=TRUE
47-
make clean
48-
pip install --upgrade --force-reinstall --no-deps .
49-
env
50-
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
51-
export MPLBACKEND=Agg
52-
export MXNET_CUDNN_AUTOTUNE_DEFAULT=0
43+
pip install --upgrade pip setuptools wheel
44+
pip install --upgrade -e .
5345
export TINY_COCO=~/.mxnet/datasets/tiny_coco
5446
export TINY_MOTORBIKE=~/.mxnet/datasets/tiny_motorbike
5547
mkdir -p $TINY_COCO/annotations

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ models. This toolkit offers four main features:
2626
4. Carefully designed APIs that greatly reduce the implementation complexity
2727
5. Community supports
2828

29+
Please also checkout [AutoGluon](https://github.com/autogluon/autogluon) if you have [image classification](https://auto.gluon.ai/stable/tutorials/multimodal/image_prediction/index.html) or [object detection](https://auto.gluon.ai/stable/tutorials/multimodal/object_detection/index.html) needs. We have built the [MultimodalPredictor](https://auto.gluon.ai/stable/tutorials/multimodal/index.html) with an improved model zoo, including [TIMM](https://github.com/rwightman/pytorch-image-models), [Huggingface](https://huggingface.co/), [MMDetection](https://github.com/open-mmlab/mmdetection) and more. With just a few lines of code, you can train and deploy high accuracy computer vision models for your application.
30+
31+
2932
# Demo
3033

3134
<div align="center">

docs/tutorials/index.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,10 @@ Apache MXNet Tutorials
33

44
Interested in getting started in a new CV area? Here are some tutorials to help get started.
55

6+
Note: For image classification or object detection tasks, feel free to explore tutorials in
7+
`AutoGluon MultiModalPredictor <https://auto.gluon.ai/stable/tutorials/multimodal/index.html>`_ with better support in PyTorch.
8+
9+
610
Image Classification
711
--------------------
812

gluoncv/torch/data/video_cls/multigrid_helper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import numpy as np
44

55
from torch.utils.data import Sampler
6-
from torch._six import int_classes as _int_classes
6+
_int_classes = int
77

88

99
__all__ = ['multiGridHelper', 'MultiGridBatchSampler']

tests/py3_mxnet_ci.yml

Lines changed: 1 addition & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -3,26 +3,10 @@ channels:
33
- conda-forge
44
- defaults
55
dependencies:
6-
- python=3.6
7-
- perl
8-
- sphinx=1.7.2
6+
- python=3.7
97
- nose
108
- coverage=4.5.4
11-
- scipy
12-
- cython
13-
- pip=20.2.4
14-
- requests==2.27.1
15-
- matplotlib
16-
- tqdm
17-
- pillow
189
- pip:
1910
- mxnet
2011
- coverage-badge
21-
- awscli
2212
- nose-timer
23-
- opencv-python
24-
- git+https://github.com/zhanghang1989/detail-api.git#subdirectory=PythonAPI
25-
- portalocker
26-
- autocfg>=0.0.6
27-
- boto3
28-
- autogluon.core==0.2.0

0 commit comments

Comments
 (0)