Commit 171f124a authored by Yuxin Wu's avatar Yuxin Wu

update sotabench setup script

parent a414092b
......@@ -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",
......
#!/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 -
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment