Skip to content

Commit 89a03e8

Browse files
vkhalidovfacebook-github-bot
authored andcommitted
adjust model zoo links
Summary: Corrected model zoo table generation script and the generated table. Added test scripts accidentally forgotten in the previous commit. Reviewed By: ppwwyyxx Differential Revision: D18004140 fbshipit-source-id: 18dfa1059645e46bf048e55769de4e05560e185e
1 parent 6f13759 commit 89a03e8

File tree

3 files changed

+61
-4
lines changed

3 files changed

+61
-4
lines changed
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
#!/bin/bash -e
2+
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
3+
4+
BIN="python train_net.py"
5+
OUTPUT="inference_test_output"
6+
NUM_GPUS=2
7+
8+
CFG_LIST=( "${@:1}" )
9+
10+
if [ ${#CFG_LIST[@]} -eq 0 ]; then
11+
CFG_LIST=( ./configs/quick_schedules/*inference_acc_test.yaml )
12+
fi
13+
14+
echo "========================================================================"
15+
echo "Configs to run:"
16+
echo "${CFG_LIST[@]}"
17+
echo "========================================================================"
18+
19+
for cfg in "${CFG_LIST[@]}"; do
20+
echo "========================================================================"
21+
echo "Running $cfg ..."
22+
echo "========================================================================"
23+
$BIN \
24+
--eval-only \
25+
--num-gpus $NUM_GPUS \
26+
--config-file "$cfg" \
27+
OUTPUT_DIR $OUTPUT
28+
rm -rf $OUTPUT
29+
done
30+
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
#!/bin/bash -e
2+
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
3+
4+
BIN="python train_net.py"
5+
OUTPUT="instant_test_output"
6+
NUM_GPUS=2
7+
8+
CFG_LIST=( "${@:1}" )
9+
if [ ${#CFG_LIST[@]} -eq 0 ]; then
10+
CFG_LIST=( ./configs/quick_schedules/*instant_test.yaml )
11+
fi
12+
13+
echo "========================================================================"
14+
echo "Configs to run:"
15+
echo "${CFG_LIST[@]}"
16+
echo "========================================================================"
17+
18+
for cfg in "${CFG_LIST[@]}"; do
19+
echo "========================================================================"
20+
echo "Running $cfg ..."
21+
echo "========================================================================"
22+
$BIN --num-gpus $NUM_GPUS --config-file "$cfg" \
23+
SOLVER.IMS_PER_BATCH $(($NUM_GPUS * 2)) \
24+
OUTPUT_DIR "$OUTPUT"
25+
rm -rf "$OUTPUT"
26+
done
27+

projects/DensePose/doc/MODEL_ZOO.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,26 +29,26 @@ All models available for download through this document are licensed under the
2929
<th valign="bottom">download</th>
3030
<!-- TABLE BODY -->
3131
<!-- ROW: densepose_rcnn_R_50_FPN_s1x -->
32-
<tr><td align="left"><a href="configs/densepose_rcnn_R_50_FPN_s1x.yaml">R_50_FPN_s1x</a></td>
32+
<tr><td align="left"><a href="../configs/densepose_rcnn_R_50_FPN_s1x.yaml">R_50_FPN_s1x</a></td>
3333
<td align="center">s1x</td>
3434
<td align="center">0.281</td>
3535
<td align="center">0.064</td>
3636
<td align="center">3.2</td>
3737
<td align="center">57.8</td>
3838
<td align="center">49.8</td>
3939
<td align="center">143908701</td>
40-
<td align="center"><a href="https://dl.fbaipublicfiles.com/DensePose//densepose_rcnn_R_50_FPN_s1x/143908701/model_final_dd99d2.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/DensePose//densepose_rcnn_R_50_FPN_s1x/143908701/metrics.json">metrics</a></td>
40+
<td align="center"><a href="https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_50_FPN_s1x/143908701/model_final_dd99d2.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_50_FPN_s1x/143908701/metrics.json">metrics</a></td>
4141
</tr>
4242
<!-- ROW: densepose_rcnn_R_101_FPN_s1x -->
43-
<tr><td align="left"><a href="configs/densepose_rcnn_R_101_FPN_s1x.yaml">R_101_FPN_s1x</a></td>
43+
<tr><td align="left"><a href="../configs/densepose_rcnn_R_101_FPN_s1x.yaml">R_101_FPN_s1x</a></td>
4444
<td align="center">s1x</td>
4545
<td align="center">0.365</td>
4646
<td align="center">0.076</td>
4747
<td align="center">4.3</td>
4848
<td align="center">59.5</td>
4949
<td align="center">51.1</td>
5050
<td align="center">143908726</td>
51-
<td align="center"><a href="https://dl.fbaipublicfiles.com/DensePose//densepose_rcnn_R_101_FPN_s1x/143908726/model_final_ad63b5.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/DensePose//densepose_rcnn_R_101_FPN_s1x/143908726/metrics.json">metrics</a></td>
51+
<td align="center"><a href="https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_101_FPN_s1x/143908726/model_final_ad63b5.pkl">model</a>&nbsp;|&nbsp;<a href="https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_101_FPN_s1x/143908726/metrics.json">metrics</a></td>
5252
</tr>
5353
</tbody></table>
5454

0 commit comments

Comments
 (0)