Skip to content

Commit afc691c

Browse files
committed
Merge: [MRCNN/PYT] Fix coco data download
2 parents 448be1b + 4e9bae8 commit afc691c

3 files changed

Lines changed: 3 additions & 8 deletions

File tree

PyTorch/Segmentation/MaskRCNN/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,8 @@ RUN pip install --upgrade --no-cache-dir pip \
2323
&& pip install --no-cache-dir \
2424
mlperf-compliance==0.0.10 \
2525
git+https://github.com/NVIDIA/dllogger \
26-
yacs
26+
yacs \
27+
dtrx
2728

2829
# Copy detectron code and build
2930
WORKDIR /workspace/object_detection

PyTorch/Segmentation/MaskRCNN/download_dataset.sh

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
DATA_DIR=$1
22

3-
wget -c https://dl.fbaipublicfiles.com/detectron/coco/coco_annotations_minival.tgz
43
wget -c http://images.cocodataset.org/zips/train2017.zip
54
wget -c http://images.cocodataset.org/zips/val2017.zip
65
wget -c http://images.cocodataset.org/annotations/annotations_trainval2017.zip
@@ -9,16 +8,12 @@ if md5sum -c hashes.md5
98
then
109
echo "DOWNLOAD PASSED"
1110
# mkdir $DATA_DIR
12-
mv coco_annotations_minival.tgz $DATA_DIR
1311
mv train2017.zip $DATA_DIR
1412
mv val2017.zip $DATA_DIR
1513
mv annotations_trainval2017.zip $DATA_DIR
1614

1715
cd $DATA_DIR
18-
dtrx --one=here coco_annotations_minival.tgz
1916
dtrx --one=here annotations_trainval2017.zip
20-
mv annotations.1/* annotations/
21-
2217
dtrx train2017.zip
2318
dtrx val2017.zip
2419

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
2d2b9d2283adb5e3b8d25eec88e65064 coco_annotations_minival.tgz
2-
bdbcd258b8721a1af3c2ae200c3d89fb train2017.zip
1+
cced6f7f71b7629ddf16f17bbcfab6b2 train2017.zip
32
442b8da7639aecaf257c1dceb8ba8c80 val2017.zip
43
f4bbac642086de4f52a3fdda2de5fa2c annotations_trainval2017.zip

0 commit comments

Comments
 (0)