prakrutpatel commited on
Commit
db9d2af
·
1 Parent(s): b844a9b

Upload 8 files

Browse files
Context RCNN/checkpoint ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ model_checkpoint_path: "model.ckpt"
2
+ all_model_checkpoint_paths: "model.ckpt"
Context RCNN/frozen_inference_graph.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6541ca8a1fc40162a5a91d48131d21e7c0c7d45bb64b4edf49c1fed0c014cdb6
3
+ size 257609643
Context RCNN/model.ckpt.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7c48fe8a6adc87d05b0db59707e6d69b676e24b6e7f2a2faf153f0a222d723b
3
+ size 316831808
Context RCNN/model.ckpt.index ADDED
Binary file (26.5 kB). View file
 
Context RCNN/model.ckpt.meta ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ecfee57731ecd6d979c4266fed7da6e488b65da7d35de5ae02be05b56d804881
3
+ size 2391425
Context RCNN/pipeline.config ADDED
@@ -0,0 +1,154 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model {
2
+ faster_rcnn {
3
+ num_classes: 1
4
+ image_resizer {
5
+ fixed_shape_resizer {
6
+ height: 640
7
+ width: 640
8
+ }
9
+ }
10
+ feature_extractor {
11
+ type: "faster_rcnn_resnet101"
12
+ first_stage_features_stride: 16
13
+ batch_norm_trainable: true
14
+ }
15
+ first_stage_anchor_generator {
16
+ grid_anchor_generator {
17
+ height_stride: 16
18
+ width_stride: 16
19
+ scales: 0.25
20
+ scales: 0.5
21
+ scales: 1.0
22
+ scales: 2.0
23
+ aspect_ratios: 0.5
24
+ aspect_ratios: 1.0
25
+ aspect_ratios: 2.0
26
+ }
27
+ }
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: 14
47
+ maxpool_kernel_size: 2
48
+ maxpool_stride: 2
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
+ share_box_across_classes: true
69
+ }
70
+ }
71
+ second_stage_post_processing {
72
+ batch_non_max_suppression {
73
+ score_threshold: 0.0
74
+ iou_threshold: 0.6
75
+ max_detections_per_class: 100
76
+ max_total_detections: 300
77
+ }
78
+ score_converter: SOFTMAX
79
+ }
80
+ second_stage_localization_loss_weight: 2.0
81
+ second_stage_classification_loss_weight: 1.0
82
+ use_matmul_crop_and_resize: true
83
+ clip_anchors_to_image: true
84
+ use_matmul_gather_in_matcher: true
85
+ use_static_balanced_label_sampler: true
86
+ use_static_shapes: true
87
+ context_config {
88
+ max_num_context_features: 2000
89
+ context_feature_length: 2057
90
+ }
91
+ }
92
+ }
93
+ train_config {
94
+ batch_size: 4
95
+ data_augmentation_options {
96
+ random_horizontal_flip {
97
+ }
98
+ }
99
+ sync_replicas: true
100
+ optimizer {
101
+ momentum_optimizer {
102
+ learning_rate {
103
+ manual_step_learning_rate {
104
+ initial_learning_rate: 0.02
105
+ schedule {
106
+ step: 1000
107
+ learning_rate: 0.0075
108
+ }
109
+ schedule {
110
+ step: 2000
111
+ learning_rate: 0.003
112
+ }
113
+ schedule {
114
+ step: 3500
115
+ learning_rate: 0.001
116
+ }
117
+ warmup: true
118
+ }
119
+ }
120
+ momentum_optimizer_value: 0.9
121
+ }
122
+ use_moving_average: false
123
+ }
124
+ gradient_clipping_by_norm: 10.0
125
+ fine_tune_checkpoint: "/content/context_rcnn_resnet101_snapshot_serengeti_2020_06_10/model.ckpt"
126
+ from_detection_checkpoint: true
127
+ num_steps: 4000
128
+ replicas_to_aggregate: 8
129
+ max_number_of_boxes: 100
130
+ unpad_groundtruth_tensors: false
131
+ use_bfloat16: true
132
+ }
133
+ train_input_reader {
134
+ label_map_path: "/content/tortoise_label_map.pbtxt"
135
+ tf_record_input_reader {
136
+ input_path: "/content/drive/MyDrive/B004_long_term_tfrecords/B004_context_rcnn_memory_train_embedding_with_context-00001-of-00001"
137
+ }
138
+ load_context_features: true
139
+ }
140
+ eval_config {
141
+ max_evals: 50
142
+ metrics_set: "coco_detection_metrics"
143
+ use_moving_averages: false
144
+ batch_size: 1
145
+ }
146
+ eval_input_reader {
147
+ label_map_path: "/content/tortoise_label_map.pbtxt"
148
+ shuffle: false
149
+ num_epochs: 1
150
+ tf_record_input_reader {
151
+ input_path: "/content/drive/MyDrive/B004_long_term_tfrecords/B004_context_rcnn_memory_test_embedding_with_context-00001-of-00001"
152
+ }
153
+ load_context_features: true
154
+ }
Context RCNN/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c08251418d1eb578b3fbb1905a64176bb50f9eb24da3576e0a836beda6f1e6c
3
+ size 257721081
Context RCNN/saved_model/variables/del.txt ADDED
File without changes