saritha5 commited on
Commit
53564bd
·
1 Parent(s): 49330e6

Upload 7 files

Browse files
.gitattributes CHANGED
@@ -32,3 +32,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
35
+ inference/model.ckpt.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
36
+ inference/model.ckpt.meta filter=lfs diff=lfs merge=lfs -text
inference/checkpoint ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ model_checkpoint_path: "model.ckpt"
2
+ all_model_checkpoint_paths: "model.ckpt"
inference/frozen_inference_graph.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e12abb4b7ff3b2da1a8a8d1233df6672d403d6e051207a95b104f5535f7307f4
3
+ size 242980390
inference/model.ckpt.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc2414288798ed90e9f627449f95c142c2e3ce6852fb793ca121023735143044
3
+ size 238479108
inference/model.ckpt.index ADDED
Binary file (43 kB). View file
 
inference/model.ckpt.meta ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c29c2b9eeb15d427808bcc7426abed960869af59e1f629937c98723ee61df9df
3
+ size 7606969
inference/pipeline.config ADDED
@@ -0,0 +1,133 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model {
2
+ faster_rcnn {
3
+ num_classes: 30
4
+ image_resizer {
5
+ keep_aspect_ratio_resizer {
6
+ min_dimension: 256
7
+ max_dimension: 640
8
+ }
9
+ }
10
+ feature_extractor {
11
+ type: "faster_rcnn_inception_resnet_v2"
12
+ first_stage_features_stride: 8
13
+ }
14
+ first_stage_anchor_generator {
15
+ grid_anchor_generator {
16
+ height_stride: 8
17
+ width_stride: 8
18
+ scales: 0.25
19
+ scales: 0.5
20
+ scales: 1.0
21
+ scales: 2.0
22
+ aspect_ratios: 0.5
23
+ aspect_ratios: 1.0
24
+ aspect_ratios: 2.0
25
+ }
26
+ }
27
+ first_stage_atrous_rate: 2
28
+ first_stage_box_predictor_conv_hyperparams {
29
+ op: CONV
30
+ regularizer {
31
+ l2_regularizer {
32
+ weight: 0.0
33
+ }
34
+ }
35
+ initializer {
36
+ truncated_normal_initializer {
37
+ stddev: 0.01
38
+ }
39
+ }
40
+ }
41
+ first_stage_nms_score_threshold: 0.0
42
+ first_stage_nms_iou_threshold: 0.7
43
+ first_stage_max_proposals: 300
44
+ first_stage_localization_loss_weight: 2.0
45
+ first_stage_objectness_loss_weight: 1.0
46
+ initial_crop_size: 17
47
+ maxpool_kernel_size: 1
48
+ maxpool_stride: 1
49
+ second_stage_box_predictor {
50
+ mask_rcnn_box_predictor {
51
+ fc_hyperparams {
52
+ op: FC
53
+ regularizer {
54
+ l2_regularizer {
55
+ weight: 0.0
56
+ }
57
+ }
58
+ initializer {
59
+ variance_scaling_initializer {
60
+ factor: 1.0
61
+ uniform: true
62
+ mode: FAN_AVG
63
+ }
64
+ }
65
+ }
66
+ use_dropout: false
67
+ dropout_keep_probability: 1.0
68
+ }
69
+ }
70
+ second_stage_post_processing {
71
+ batch_non_max_suppression {
72
+ score_threshold: 0.3
73
+ iou_threshold: 0.6
74
+ max_detections_per_class: 100
75
+ max_total_detections: 100
76
+ }
77
+ score_converter: SOFTMAX
78
+ }
79
+ second_stage_localization_loss_weight: 2.0
80
+ second_stage_classification_loss_weight: 1.0
81
+ }
82
+ }
83
+ train_config {
84
+ batch_size: 1
85
+ data_augmentation_options {
86
+ random_horizontal_flip {
87
+ }
88
+ }
89
+ optimizer {
90
+ momentum_optimizer {
91
+ learning_rate {
92
+ manual_step_learning_rate {
93
+ initial_learning_rate: 0.0003
94
+ schedule {
95
+ step: 900000
96
+ learning_rate: 3e-05
97
+ }
98
+ schedule {
99
+ step: 1200000
100
+ learning_rate: 3e-06
101
+ }
102
+ }
103
+ }
104
+ momentum_optimizer_value: 0.9
105
+ }
106
+ use_moving_average: false
107
+ }
108
+ gradient_clipping_by_norm: 10.0
109
+ fine_tune_checkpoint: "gs://te_object_detection/pre_trained_model/faster_rcnn_inception_resnet_v2_atrous_coco_2018_01_28/model.ckpt"
110
+ from_detection_checkpoint: false
111
+ num_steps: 40000
112
+ load_all_detection_checkpoint_vars: true
113
+ fine_tune_checkpoint_type: "detection"
114
+ }
115
+ train_input_reader {
116
+ label_map_path: "gs://te_object_detection/ytag/tf_label_map.pbtxt"
117
+ tf_record_input_reader {
118
+ input_path: "gs://te_object_detection/ytag/495-images-28-oct-YB-train.record"
119
+ }
120
+ }
121
+ eval_config {
122
+ num_examples: 8000
123
+ max_evals: 10
124
+ use_moving_averages: false
125
+ }
126
+ eval_input_reader {
127
+ label_map_path: "gs://te_object_detection/ytag/tf_label_map.pbtxt"
128
+ shuffle: true
129
+ num_readers: 1
130
+ tf_record_input_reader {
131
+ input_path: "gs://te_object_detection/ytag/80-images-28-oct-YB-test.record"
132
+ }
133
+ }
inference/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76fba7a987d3dd59d66575c6f405bdbccb87aba001750df1f6466456c6a75d19
3
+ size 243431983