23b719w commited on
Commit
6309e38
1 Parent(s): d7c9a6b

Upload 7 files

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: 1704306271.0134602
4
+ last_preserved_timestamp: 1704306269.6748242
checkpoint/ckpt-0.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:575647cb4e489331ef352a97424347725da32f55e7e1c24cd8d71915306cc473
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: 10
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: 25000
80
+ warmup_learning_rate: 0.0003
81
+ warmup_steps: 2000
82
+ }
83
+ }
84
+ epsilon: 1e-07
85
+ }
86
+ use_moving_average: false
87
+ }
88
+ fine_tune_checkpoint: "/home/ubuntu/assignment2_project/pretrained_models/centernet_resnet50_v1_fpn_512x512_coco17_tpu-8/checkpoint/ckpt-0"
89
+ num_steps: 25000
90
+ max_number_of_boxes: 100
91
+ unpad_groundtruth_tensors: false
92
+ fine_tune_checkpoint_type: "detection"
93
+ use_bfloat16: false
94
+ fine_tune_checkpoint_version: V2
95
+ }
96
+ train_input_reader {
97
+ label_map_path: "/home/ubuntu/assignment2_project/data/label_map.pbtxt"
98
+ tf_record_input_reader {
99
+ input_path: "/home/ubuntu/assignment2_project/data/train.record"
100
+ }
101
+ }
102
+ eval_config {
103
+ metrics_set: "coco_detection_metrics"
104
+ use_moving_averages: false
105
+ }
106
+ eval_input_reader {
107
+ label_map_path: "/home/ubuntu/assignment2_project/data/label_map.pbtxt"
108
+ shuffle: false
109
+ num_epochs: 1
110
+ tf_record_input_reader {
111
+ input_path: "/home/ubuntu/assignment2_project/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:dcbd159613f7c09469db50afad0e65a9bfbfc889f4de82c35f8d05f1cace1225
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:4832eb86ea4516c10b4478e7a9d5df916d3705a2c25cb6046b6eb38ee771029f
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:2a19f5a36f9b63d312010668c9a2d7c1e96f5535540028bdc60f3f7d5e7ca81b
3
+ size 20951