Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
171f124a
Commit
171f124a
authored
Sep 29, 2019
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update sotabench setup script
parent
a414092b
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
4 deletions
+9
-4
sotabench.py
sotabench.py
+5
-0
sotabench_setup.sh
sotabench_setup.sh
+4
-4
No files found.
sotabench.py
View file @
171f124a
...
...
@@ -6,6 +6,7 @@ import tqdm
from
tensorpack.predict
import
OfflinePredictor
,
PredictConfig
from
tensorpack.tfutils
import
SmartInit
from
tensorpack.utils.fs
import
download
from
sotabencheval.utils
import
is_server
from
sotabencheval.object_detection
import
COCOEvaluator
...
...
@@ -76,6 +77,10 @@ def evaluate_rcnn(model_name, paper_arxiv_id, cfg_list, model_file):
evaluator
.
save
()
download
(
"http://models.tensorpack.com/FasterRCNN/COCO-MaskRCNN-R101FPN9xGNCasAugScratch.npz"
,
"./"
,
expect_size
=
355680386
)
evaluate_rcnn
(
"Mask R-CNN (ResNet-101-FPN, GN, Cascade)"
,
"1811.08883"
,
...
...
sotabench_setup.sh
View file @
171f124a
#!/bin/bash
pip
install
-e
.
.
/workspace/venv/bin/activate
wget http://models.tensorpack.com/FasterRCNN/COCO-MaskRCNN-R101FPN9xGNCasAugScratch.npz
pip
install
-e
.
cd
./.data/vision/coco
unzip annotations_trainval2017.zip
unzip val2017.zip
python
-c
'import zipfile; zipfile.ZipFile("annotations_trainval2017.zip").extractall()'
python
-c
'import zipfile; zipfile.ZipFile("val2017.zip").extractall()'
cd
-
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