add requirements.txt, update config

9027859a · Shaoshuai Shi · 98281d05 · 9027859a · 9027859a
Commit 9027859a authored Jun 24, 2020 by Shaoshuai Shi
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 2 deletions

requirements.txt requirements.txt +8 -0

tools/cfgs/kitti_models/PartA2.yaml tools/cfgs/kitti_models/PartA2.yaml +2 -2

No files found.
--- a/requirements.txt
+++ b/requirements.txt
+numpy
+torch>=1.1
+numba
+tensorboardX
+easydict
+pyyaml
+scikit-image
+tqdm
--- a/tools/cfgs/kitti_models/PartA2.yaml
+++ b/tools/cfgs/kitti_models/PartA2.yaml
@@ -46,14 +46,14 @@ MODEL:
            {
                'anchor_sizes': [[0.8, 0.6, 1.73]],
                'anchor_rotations': [0, 1.57],
-                'anchor_bottom_heights': [-0.6],
+                'anchor_bottom_heights': [-1.78],
                'align_center': False,
                'feature_map_stride': 8
            },
            {
                'anchor_sizes': [[1.76, 0.6, 1.73]],
                'anchor_rotations': [0, 1.57],
-                'anchor_bottom_heights': [-0.6],
+                'anchor_bottom_heights': [-1.78],
                'align_center': False,
                'feature_map_stride': 8
            }