sokonana commited on
Commit
745da8e
·
1 Parent(s): df8cc96

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ checkpoint/ckpt-0.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
checkpoint/checkpoint ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ model_checkpoint_path: "ckpt-0"
2
+ all_model_checkpoint_paths: "ckpt-0"
3
+ all_model_checkpoint_timestamps: 1703423270.1737576
4
+ last_preserved_timestamp: 1703423268.8147209
checkpoint/ckpt-0.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5f6aecd99df93e077216c0f152600baaf27e95762bf91bf7ee3d205cdb8dea6
3
+ size 101303656
checkpoint/ckpt-0.index ADDED
Binary file (23.5 kB). View file
 
pipeline.config ADDED
@@ -0,0 +1,113 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model {
2
+ center_net {
3
+ num_classes: 2
4
+ feature_extractor {
5
+ type: "resnet_v1_50_fpn"
6
+ }
7
+ image_resizer {
8
+ keep_aspect_ratio_resizer {
9
+ min_dimension: 512
10
+ max_dimension: 512
11
+ pad_to_max_dimension: true
12
+ }
13
+ }
14
+ object_detection_task {
15
+ task_loss_weight: 1.0
16
+ offset_loss_weight: 1.0
17
+ scale_loss_weight: 0.1
18
+ localization_loss {
19
+ l1_localization_loss {
20
+ }
21
+ }
22
+ }
23
+ object_center_params {
24
+ object_center_loss_weight: 1.0
25
+ classification_loss {
26
+ penalty_reduced_logistic_focal_loss {
27
+ alpha: 2.0
28
+ beta: 4.0
29
+ }
30
+ }
31
+ min_box_overlap_iou: 0.7
32
+ max_box_predictions: 100
33
+ }
34
+ }
35
+ }
36
+ train_config {
37
+ batch_size: 16
38
+ data_augmentation_options {
39
+ random_horizontal_flip {
40
+ }
41
+ }
42
+ data_augmentation_options {
43
+ random_crop_image {
44
+ min_aspect_ratio: 0.5
45
+ max_aspect_ratio: 1.7
46
+ random_coef: 0.25
47
+ }
48
+ }
49
+ data_augmentation_options {
50
+ random_adjust_hue {
51
+ }
52
+ }
53
+ data_augmentation_options {
54
+ random_adjust_contrast {
55
+ }
56
+ }
57
+ data_augmentation_options {
58
+ random_adjust_saturation {
59
+ }
60
+ }
61
+ data_augmentation_options {
62
+ random_adjust_brightness {
63
+ }
64
+ }
65
+ data_augmentation_options {
66
+ random_absolute_pad_image {
67
+ max_height_padding: 200
68
+ max_width_padding: 200
69
+ pad_color: 0.0
70
+ pad_color: 0.0
71
+ pad_color: 0.0
72
+ }
73
+ }
74
+ optimizer {
75
+ adam_optimizer {
76
+ learning_rate {
77
+ cosine_decay_learning_rate {
78
+ learning_rate_base: 0.001
79
+ total_steps: 50000
80
+ warmup_learning_rate: 0.00025
81
+ warmup_steps: 5000
82
+ }
83
+ }
84
+ epsilon: 1e-07
85
+ }
86
+ use_moving_average: false
87
+ }
88
+ fine_tune_checkpoint: "/home/ubuntu/it107_assignment/pretrained_models/centernet_resnet50_v1_fpn_512x512_coco17_tpu-8/checkpoint/ckpt-0"
89
+ num_steps: 50000
90
+ max_number_of_boxes: 100
91
+ unpad_groundtruth_tensors: false
92
+ fine_tune_checkpoint_type: "detection"
93
+ fine_tune_checkpoint_version: V2
94
+ }
95
+ train_input_reader {
96
+ label_map_path: "/home/ubuntu/it107_assignment/data/label_map.pbtxt"
97
+ tf_record_input_reader {
98
+ input_path: "/home/ubuntu/it107_assignment/data/train.record"
99
+ }
100
+ }
101
+ eval_config {
102
+ metrics_set: "coco_detection_metrics"
103
+ use_moving_averages: false
104
+ batch_size: 1
105
+ }
106
+ eval_input_reader {
107
+ label_map_path: "/home/ubuntu/it107_assignment/data/label_map.pbtxt"
108
+ shuffle: false
109
+ num_epochs: 1
110
+ tf_record_input_reader {
111
+ input_path: "/home/ubuntu/it107_assignment/data/val.record"
112
+ }
113
+ }
saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:951134bd4325908a47a8a7da282cde8d434c1a29bba12673aa7c06fd7ba4555f
3
+ size 9097925
saved_model/variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:047cd7b67d65d5977fbb6a05a0bf4fff0370f82402d5f4a2e5f92afd7b9fa68b
3
+ size 101386553
saved_model/variables/variables.index ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:931e0ce42a7251cd77012d837fc3e30f28b47fd101bcc4c0818033801fad38c2
3
+ size 20951