francepfl commited on
Commit
c401174
1 Parent(s): 727d0f8

End of training

Browse files
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ step_*
2
+ epoch_*
all_results.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "test_map": 0.1158,
3
+ "test_map_50": 0.2001,
4
+ "test_map_75": 0.1156,
5
+ "test_map_small": 0.0255,
6
+ "test_map_medium": 0.0297,
7
+ "test_map_large": 0.1237,
8
+ "test_mar_1": 0.1176,
9
+ "test_mar_10": 0.223,
10
+ "test_mar_100": 0.2509,
11
+ "test_mar_small": 0.0739,
12
+ "test_mar_medium": 0.1302,
13
+ "test_mar_large": 0.3261,
14
+ "test_map_Coverall": 0.4697,
15
+ "test_mar_100_Coverall": 0.6933,
16
+ "test_map_Face_Shield": 0.0,
17
+ "test_mar_100_Face_Shield": 0.0,
18
+ "test_map_Gloves": 0.0324,
19
+ "test_mar_100_Gloves": 0.259,
20
+ "test_map_Goggles": 0.0,
21
+ "test_mar_100_Goggles": 0.0,
22
+ "test_map_Mask": 0.0768,
23
+ "test_mar_100_Mask": 0.3019
24
+ }
config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "hustvl/yolos-base",
3
+ "architectures": [
4
+ "YolosForObjectDetection"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "auxiliary_loss": false,
8
+ "bbox_cost": 5,
9
+ "bbox_loss_coefficient": 5,
10
+ "class_cost": 1,
11
+ "eos_coefficient": 0.1,
12
+ "giou_cost": 2,
13
+ "giou_loss_coefficient": 2,
14
+ "hidden_act": "gelu",
15
+ "hidden_dropout_prob": 0.0,
16
+ "hidden_size": 768,
17
+ "id2label": {
18
+ "0": "Coverall",
19
+ "1": "Face_Shield",
20
+ "2": "Gloves",
21
+ "3": "Goggles",
22
+ "4": "Mask"
23
+ },
24
+ "image_size": [
25
+ 800,
26
+ 1344
27
+ ],
28
+ "initializer_range": 0.02,
29
+ "intermediate_size": 3072,
30
+ "label2id": {
31
+ "Coverall": 0,
32
+ "Face_Shield": 1,
33
+ "Gloves": 2,
34
+ "Goggles": 3,
35
+ "Mask": 4
36
+ },
37
+ "layer_norm_eps": 1e-12,
38
+ "model_type": "yolos",
39
+ "num_attention_heads": 12,
40
+ "num_channels": 3,
41
+ "num_detection_tokens": 100,
42
+ "num_hidden_layers": 12,
43
+ "patch_size": 16,
44
+ "qkv_bias": true,
45
+ "torch_dtype": "float32",
46
+ "transformers_version": "4.42.0.dev0",
47
+ "use_mid_position_embeddings": true
48
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b30bd5309441bfd8fed6b1949dd8e9eb152e45bfd0632c4c36cb0de88ef0bfb
3
+ size 510953464
object_detection_no_trainer/1716384293.957731/events.out.tfevents.1716384293.drivelm-interactive-0-0.4299.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b9171f70ac7754af2e478cab5a487ad055a7fb6589ce9ec6d8e9f2ba0645673
3
+ size 1621
object_detection_no_trainer/1716384293.9730523/hparams.yml ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ cache_dir: null
5
+ checkpointing_steps: epoch
6
+ dataloader_num_workers: 4
7
+ dataset_name: cppe-5
8
+ gradient_accumulation_steps: 1
9
+ hub_model_id: null
10
+ hub_token: null
11
+ ignore_mismatched_sizes: true
12
+ image_square_size: 600
13
+ learning_rate: 5.0e-05
14
+ lr_scheduler_type: linear
15
+ max_train_steps: 10700
16
+ model_name_or_path: hustvl/yolos-base
17
+ num_train_epochs: 100
18
+ num_warmup_steps: 0
19
+ output_dir: detr-resnet-50-finetuned
20
+ per_device_eval_batch_size: 8
21
+ per_device_train_batch_size: 8
22
+ push_to_hub: true
23
+ report_to: all
24
+ resume_from_checkpoint: null
25
+ seed: null
26
+ train_val_split: 0.15
27
+ trust_remote_code: false
28
+ use_auth_token: false
29
+ with_tracking: true
object_detection_no_trainer/events.out.tfevents.1716384283.drivelm-interactive-0-0.4299.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc201cf341c225e47c8f3c81a78e747b3190ee2769fd8460d62f9ae3ea25ce12
3
+ size 88
preprocessor_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "annotations",
5
+ "return_segmentation_masks",
6
+ "masks_path",
7
+ "do_resize",
8
+ "size",
9
+ "resample",
10
+ "do_rescale",
11
+ "rescale_factor",
12
+ "do_normalize",
13
+ "image_mean",
14
+ "image_std",
15
+ "do_convert_annotations",
16
+ "do_pad",
17
+ "pad_size",
18
+ "format",
19
+ "return_tensors",
20
+ "data_format",
21
+ "input_data_format"
22
+ ],
23
+ "do_convert_annotations": true,
24
+ "do_normalize": true,
25
+ "do_pad": false,
26
+ "do_rescale": true,
27
+ "do_resize": false,
28
+ "format": "coco_detection",
29
+ "image_mean": [
30
+ 0.485,
31
+ 0.456,
32
+ 0.406
33
+ ],
34
+ "image_processor_type": "YolosImageProcessor",
35
+ "image_std": [
36
+ 0.229,
37
+ 0.224,
38
+ 0.225
39
+ ],
40
+ "pad_size": null,
41
+ "resample": 2,
42
+ "rescale_factor": 0.00392156862745098,
43
+ "size": {
44
+ "longest_edge": 600
45
+ }
46
+ }