Add TensionTRAC 3D HoD intra-cross-sem checkpoint
Browse files
.gitattributes
CHANGED
|
@@ -40,3 +40,4 @@ TensionTRAC_2d_intra_cross/checkpoint_best.pyth filter=lfs diff=lfs merge=lfs -t
|
|
| 40 |
TensionTRAC_2d_intra_cross_sem/checkpoint_best.pyth filter=lfs diff=lfs merge=lfs -text
|
| 41 |
TensionTRAC_2d_intra_sem/checkpoint_best.pyth filter=lfs diff=lfs merge=lfs -text
|
| 42 |
TensionTRAC_2d_sem/checkpoint_best.pyth filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 40 |
TensionTRAC_2d_intra_cross_sem/checkpoint_best.pyth filter=lfs diff=lfs merge=lfs -text
|
| 41 |
TensionTRAC_2d_intra_sem/checkpoint_best.pyth filter=lfs diff=lfs merge=lfs -text
|
| 42 |
TensionTRAC_2d_sem/checkpoint_best.pyth filter=lfs diff=lfs merge=lfs -text
|
| 43 |
+
TensionTRAC_3d_HoD_intra_cross_sem/checkpoint_best.pyth filter=lfs diff=lfs merge=lfs -text
|
TensionTRAC_3d_HoD_intra_cross_sem/TensionTRAC_3d_HoD_intra_cross_sem.yaml
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
TASK: few_shot
|
| 2 |
+
DATA:
|
| 3 |
+
NUM_FRAMES: 8
|
| 4 |
+
TRAIN_CROP_SIZE: 224
|
| 5 |
+
MEAN: [0.5, 0.5, 0.5]
|
| 6 |
+
STD: [0.5, 0.5, 0.5]
|
| 7 |
+
REVERSE_INPUT_CHANNEL: True
|
| 8 |
+
FEW_SHOT:
|
| 9 |
+
TRAJ_REWEIGHT:
|
| 10 |
+
ENABLE: True
|
| 11 |
+
POINT_INFO:
|
| 12 |
+
ENABLE: True
|
| 13 |
+
NUM_POINTS_TO_SAMPLE: 256
|
| 14 |
+
POINT_DIM : 3
|
| 15 |
+
HOD_ORIENTATION_HIST_TYPE: '3d_interplane'
|
| 16 |
+
USE_PT_QUERY_MASK: True
|
| 17 |
+
HOD:
|
| 18 |
+
NUM_BINS: 32
|
| 19 |
+
TEMPORAL_PYRAMID: False
|
| 20 |
+
MODEL:
|
| 21 |
+
MODEL_NAME: Pointformer
|
| 22 |
+
ARCH: slow
|
| 23 |
+
NUM_CLASSES: 64
|
| 24 |
+
FEAT_EXTRACTOR: dino
|
| 25 |
+
DINO_CONFIG: dinov2_vitb14
|
| 26 |
+
LOAD_DINO_ON_INIT: True
|
| 27 |
+
USE_DEPTH: True
|
| 28 |
+
USE_PRECOMPUTED_DEPTH: False
|
| 29 |
+
DEPTH:
|
| 30 |
+
MODEL: "dpt"
|
| 31 |
+
ENCODER: "vitb"
|
| 32 |
+
INPUT_SIZE: 518
|
| 33 |
+
MAX_RES: 1280
|
| 34 |
+
FP32: False
|
| 35 |
+
IS_METRICS: False
|
| 36 |
+
DEPTH_EMBEDDER:
|
| 37 |
+
INPUT_DEPTH_MAP_DIM: 256
|
| 38 |
+
OUTPUT_DEPTH_EMBED_DIM: 32
|
| 39 |
+
FUSION:
|
| 40 |
+
TYPE: add
|
| 41 |
+
OUTPUT_DIM: None
|
| 42 |
+
MOTION_MODULE:
|
| 43 |
+
USE_HOD_MOTION_MODULE: True
|
| 44 |
+
USE_CROSS_MOTION_MODULE: True
|
| 45 |
+
INNER_MODULE_TYPE: hod
|
| 46 |
+
INTER_MODULE_TYPE: legacy_cross
|
| 47 |
+
OUT_FEATURE_DIM: 768
|
| 48 |
+
INNER_HIDDEN_DIM: 128
|
| 49 |
+
INTER_HIDDEN_DIM: 128
|
| 50 |
+
INNER_KERNEL_SIZE: 3
|
| 51 |
+
K_NEIGHBORS: 8
|
| 52 |
+
DROPOUT: 0.1
|
| 53 |
+
SMOOTH_Z: True
|
TensionTRAC_3d_HoD_intra_cross_sem/checkpoint_best.pyth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf9e8ffe1cb01af3c887c9b9512c713f3ec108bf718e96ca2cc48edacc5be8b2
|
| 3 |
+
size 668374123
|