Mhara commited on
Commit
fccd3a3
·
verified ·
1 Parent(s): 8964804

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