caball21 commited on
Commit
755ff1c
·
verified ·
1 Parent(s): c29e78b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -26
config.json CHANGED
@@ -1,28 +1,14 @@
1
  {
2
- "trainer": {
3
- "max_epochs": 50,
4
- "log_every_n_steps": 10
5
- },
6
- "experiment_log_dir": "/content/drive/MyDrive/glove_labelling/training_output",
7
- "data": {
8
- "train": {
9
- "_target_": "training.dataset.sam2_datasets.TorchTrainMixedDataset",
10
- "phases_per_epoch": 100,
11
- "batch_sizes": [2],
12
- "datasets": [
13
- {
14
- "_target_": "training.dataset.vos_dataset.VOSDataset",
15
- "training": true,
16
- "video_dataset": {
17
- "_target_": "training.dataset.vos_dataset.VideoDatasetFromJSON",
18
- "root_dir": "/content/drive/MyDrive/glove_labelling/final_train_data",
19
- "json_paths": [
20
- "/content/drive/MyDrive/glove_labelling/final_train_data"
21
- ]
22
- }
23
- }
24
- ]
25
- }
26
- },
27
- "model": "sam2.1_hiera_l"
28
  }
 
1
  {
2
+ "model_type": "custom-sam2",
3
+ "architecture": "SAM2Hierarchical",
4
+ "image_size": [720, 1280],
5
+ "num_classes": 6,
6
+ "class_labels": [
7
+ "glove_outline",
8
+ "webbing",
9
+ "thumb",
10
+ "palm_pocket",
11
+ "hand",
12
+ "glove_exterior"
13
+ ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14
  }