Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
SOLOv2-pytorch
Commits
89005895
Unverified
Commit
89005895
authored
Apr 04, 2020
by
Xinlong Wang
Committed by
GitHub
Apr 04, 2020
Browse files
Update README.md
parent
c47568a4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
4 deletions
+12
-4
README.md
README.md
+12
-4
No files found.
README.md
View file @
89005895
...
...
@@ -44,8 +44,8 @@ Decoupled_SOLO_R101_3x | Yes | 92ms | 37.9 | [download](https://cloudstor.aarnet
Model | Multi-scale training | Testing time / im | AP (minival) | Link
--- |:---:|:---:|:---:|:---:
D
ECOUPLED
_SOLO_L
IGHT
_R50_3x | Yes | 29ms | 33.0 |
[
download
](
https://cloudstor.aarnet.edu.au/plus/s/d0zuZgCnAjeYvod/download
)
D
ECOUPLED
_SOLO_L
IGHT
_DCN_R50_3x | Yes | 36ms | 35.0 |
[
download
](
https://cloudstor.aarnet.edu.au/plus/s/QvWhOTmCA5pFj6E/download
)
D
ecoupled
_SOLO_L
ight
_R50_3x | Yes | 29ms | 33.0 |
[
download
](
https://cloudstor.aarnet.edu.au/plus/s/d0zuZgCnAjeYvod/download
)
D
ecoupled
_SOLO_L
ight
_DCN_R50_3x | Yes | 36ms | 35.0 |
[
download
](
https://cloudstor.aarnet.edu.au/plus/s/QvWhOTmCA5pFj6E/download
)
## Usage
...
...
@@ -66,17 +66,25 @@ Once the installation is done, you can download the provided models and use [inf
python tools/train.py configs/solo/solo_r50_fpn_8gpu_1x.py
### Testing
# multi-gpu testing
./tools/dist_test.sh ${CONFIG_FILE} ${CHECKPOINT_FILE} ${GPU_NUM} --show --out ${OUTPUT_FILE} --eval segm
Example:
./tools/dist_test.sh configs/solo/solo_r50_fpn_8gpu_1x.py SOLO_R50_1x.pth 8 --show --out results_solo.pkl --eval segm
# single-gpu testing
python tools/test_ins.py ${CONFIG_FILE} ${CHECKPOINT_FILE} --show --out ${OUTPUT_FILE} --eval segm
Example:
python tools/test_ins.py configs/solo/solo_r50_fpn_8gpu_1x.py SOLO_R50_FPN_1x.pth --show --out results_solo.pkl --eval segm
python tools/test_ins.py configs/solo/solo_r50_fpn_8gpu_1x.py SOLO_R50_1x.pth --show --out results_solo.pkl --eval segm
### Visualization
python tools/test_ins_vis.py ${CONFIG_FILE} ${CHECKPOINT_FILE} --show --save_dir ${SAVE_DIR}
Example:
python tools/test_ins_vis.py configs/solo/solo_r50_fpn_8gpu_1x.py SOLO_R50_
FPN_
1x.pth --show --save_dir work_dirs/vis_solo
python tools/test_ins_vis.py configs/solo/solo_r50_fpn_8gpu_1x.py SOLO_R50_1x.pth --show --save_dir work_dirs/vis_solo
## Contributing to the project
Any pull requests or issues are welcome.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment