File tree Expand file tree Collapse file tree 5 files changed +13
-11
lines changed
mmdeploy/codebase/mmcls/models/classifiers Expand file tree Collapse file tree 5 files changed +13
-11
lines changed Original file line number Diff line number Diff line change @@ -86,7 +86,7 @@ commands:
86
86
python -m mim install -r requirements/codebases.txt
87
87
python -m pip install -r requirements/tests.txt
88
88
python -m pip install -r requirements/runtime.txt
89
- python -m pip install -U numpy
89
+ python -m pip install -U numpy clip
90
90
cd ..
91
91
perform_model_converter_ut :
92
92
steps :
@@ -147,7 +147,7 @@ jobs:
147
147
docker exec mmdeploy mim install -r mmdeploy/requirements/codebases.txt
148
148
docker exec mmdeploy pip install -r mmdeploy/requirements/tests.txt
149
149
docker exec mmdeploy pip install -r mmdeploy/requirements/runtime.txt
150
- docker exec mmdeploy pip install -U numpy
150
+ docker exec mmdeploy pip install -U numpy clip
151
151
- run :
152
152
name : Perform Model Converter unittests
153
153
command : |
Original file line number Diff line number Diff line change 48
48
python -m pip install -r requirements.txt -r requirements/backends.txt
49
49
python -m mim install "mmcv>=2.0.0rc1"
50
50
python -m mim install -r requirements/codebases.txt
51
- python -m pip install -U numpy
51
+ python -m pip install -U numpy clip
52
52
python -m pip list
53
53
- name : Build and install
54
54
run : |
@@ -142,7 +142,7 @@ jobs:
142
142
python -m pip install -r requirements.txt -r requirements/backends.txt
143
143
python -m mim install "mmcv>=2.0.0rc1"
144
144
CFLAGS=$CFLAGS python -m mim install -r requirements/codebases.txt
145
- python -m pip install -U pycuda numpy
145
+ python -m pip install -U pycuda numpy clip
146
146
python -m pip list
147
147
- name : Build and install
148
148
run : |
@@ -190,7 +190,7 @@ jobs:
190
190
python -m pip install -r requirements.txt -r requirements/backends.txt
191
191
python -m mim install "mmcv>=2.0.0rc1"
192
192
python -m mim install -r requirements/codebases.txt
193
- python -m pip install -U pycuda numpy
193
+ python -m pip install -U pycuda numpy clip
194
194
python -m pip list
195
195
- name : Build and install
196
196
run : |
Original file line number Diff line number Diff line change @@ -30,6 +30,7 @@ def base_classifier__forward(
30
30
Returns:
31
31
return a list of :obj:`mmengine.BaseDataElement`.
32
32
"""
33
- feats = self .extract_feat (batch_inputs )
34
- output = self .head (feats )
33
+ output = self .extract_feat (batch_inputs )
34
+ if self .head is not None :
35
+ output = self .head (output )
35
36
return output
Original file line number Diff line number Diff line change 1
1
mmcls>=1.0.0rc2
2
- mmdet
@ git+https://github.com/open-mmlab/[email protected]
3
- mmdet3d
@ git+https://github.com/open-mmlab/[email protected]
4
- mmedit
@ git+https://github.com/open-mmlab/[email protected]
5
- mmocr
@ git+https://github.com/open-mmlab/[email protected]
2
+ mmdet>=3.0.0rc4
3
+ mmdet3d>=1.1.0rc2
4
+ mmedit>=1.0.0rc2
5
+ mmocr>=1.0.0rc4
6
6
mmpose>=1.0.0rc0
7
7
mmsegmentation>=1.0.0rc0
Original file line number Diff line number Diff line change 6
6
multiprocess
7
7
numpy
8
8
onnx>=1.8.0
9
+ prettytable
9
10
protobuf<=3.20.1
10
11
six
11
12
terminaltables
You can’t perform that action at this time.
0 commit comments