File tree Expand file tree Collapse file tree
PyTorch/Segmentation/MaskRCNN Expand file tree Collapse file tree Original file line number Diff line number Diff 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
2930WORKDIR /workspace/object_detection
Original file line number Diff line number Diff line change 11DATA_DIR=$1
22
3- wget -c https://dl.fbaipublicfiles.com/detectron/coco/coco_annotations_minival.tgz
43wget -c http://images.cocodataset.org/zips/train2017.zip
54wget -c http://images.cocodataset.org/zips/val2017.zip
65wget -c http://images.cocodataset.org/annotations/annotations_trainval2017.zip
@@ -9,16 +8,12 @@ if md5sum -c hashes.md5
98then
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
Original file line number Diff line number Diff line change 1- 2d2b9d2283adb5e3b8d25eec88e65064 coco_annotations_minival.tgz
2- bdbcd258b8721a1af3c2ae200c3d89fb train2017.zip
1+ cced6f7f71b7629ddf16f17bbcfab6b2 train2017.zip
32442b8da7639aecaf257c1dceb8ba8c80 val2017.zip
43f4bbac642086de4f52a3fdda2de5fa2c annotations_trainval2017.zip
You can’t perform that action at this time.
0 commit comments