prakrutpatel commited on
Commit
78e7a70
·
1 Parent(s): 30cd121

Upload 8 files

Browse files
Faster RCNN/checkpoint ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ model_checkpoint_path: "model.ckpt"
2
+ all_model_checkpoint_paths: "model.ckpt"
Faster RCNN/frozen_inference_graph.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:599f990344c0878c98f6271d7fb6430a78c5a606d7a147ed2190349adbac5790
3
+ size 190353156
Faster RCNN/model.ckpt.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22b80339e8748126e50ad03ab20cb95e75e70515ea7be612d08f13feb0b4b95b
3
+ size 249444416
Faster RCNN/model.ckpt.index ADDED
Binary file (25.7 kB). View file
 
Faster RCNN/model.ckpt.meta ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:456d23fbcf72ab395f0cf31d036765c1f2b973318d22f25b94c8b391312c9117
3
+ size 2655588
Faster RCNN/pipeline.config ADDED
@@ -0,0 +1,137 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model {
2
+ faster_rcnn {
3
+ num_classes: 1
4
+ image_resizer {
5
+ keep_aspect_ratio_resizer {
6
+ min_dimension: 600
7
+ max_dimension: 1024
8
+ }
9
+ }
10
+ feature_extractor {
11
+ type: "faster_rcnn_resnet101"
12
+ first_stage_features_stride: 16
13
+ }
14
+ first_stage_anchor_generator {
15
+ grid_anchor_generator {
16
+ height_stride: 16
17
+ width_stride: 16
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_box_predictor_conv_hyperparams {
28
+ op: CONV
29
+ regularizer {
30
+ l2_regularizer {
31
+ weight: 0.0
32
+ }
33
+ }
34
+ initializer {
35
+ truncated_normal_initializer {
36
+ stddev: 0.01
37
+ }
38
+ }
39
+ }
40
+ first_stage_nms_score_threshold: 0.0
41
+ first_stage_nms_iou_threshold: 0.7
42
+ first_stage_max_proposals: 300
43
+ first_stage_localization_loss_weight: 2.0
44
+ first_stage_objectness_loss_weight: 1.0
45
+ initial_crop_size: 14
46
+ maxpool_kernel_size: 2
47
+ maxpool_stride: 2
48
+ second_stage_box_predictor {
49
+ mask_rcnn_box_predictor {
50
+ fc_hyperparams {
51
+ op: FC
52
+ regularizer {
53
+ l2_regularizer {
54
+ weight: 0.0
55
+ }
56
+ }
57
+ initializer {
58
+ variance_scaling_initializer {
59
+ factor: 1.0
60
+ uniform: true
61
+ mode: FAN_AVG
62
+ }
63
+ }
64
+ }
65
+ use_dropout: false
66
+ dropout_keep_probability: 1.0
67
+ }
68
+ }
69
+ second_stage_post_processing {
70
+ batch_non_max_suppression {
71
+ score_threshold: 0.0
72
+ iou_threshold: 0.6
73
+ max_detections_per_class: 100
74
+ max_total_detections: 300
75
+ }
76
+ score_converter: SOFTMAX
77
+ }
78
+ second_stage_localization_loss_weight: 2.0
79
+ second_stage_classification_loss_weight: 1.0
80
+ output_final_box_features: true
81
+ }
82
+ }
83
+ train_config {
84
+ batch_size: 4
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.02
94
+ schedule {
95
+ step: 1000
96
+ learning_rate: 0.0075
97
+ }
98
+ schedule {
99
+ step: 2000
100
+ learning_rate: 0.003
101
+ }
102
+ schedule {
103
+ step: 3500
104
+ learning_rate: 0.0010000011
105
+ }
106
+ }
107
+ }
108
+ momentum_optimizer_value: 0.9
109
+ }
110
+ use_moving_average: false
111
+ }
112
+ gradient_clipping_by_norm: 10.0
113
+ fine_tune_checkpoint: "/content/faster_rcnn_resnet101_snapshot_serengeti_2020_06_10/model.ckpt"
114
+ from_detection_checkpoint: true
115
+ load_all_detection_checkpoint_vars: true
116
+ fine_tune_checkpoint_type: "detection"
117
+ use_bfloat16: false
118
+ }
119
+ train_input_reader {
120
+ label_map_path: "/content/tortoise_label_map.pbtxt"
121
+ tf_record_input_reader {
122
+ input_path: "/content/drive/MyDrive/B004_context_rcnn_memory_train"
123
+ }
124
+ }
125
+ eval_config {
126
+ metrics_set: "coco_detection_metrics"
127
+ use_moving_averages: false
128
+ }
129
+ eval_input_reader {
130
+ label_map_path: "/content/tortoise_label_map.pbtxt"
131
+ shuffle: false
132
+ num_epochs: 1
133
+ tf_record_input_reader {
134
+ input_path: "/content/drive/MyDrive/B004_context_rcnn_memory_test"
135
+ }
136
+ sample_1_of_n_examples: 1000
137
+ }
Faster RCNN/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:140c7bb676223d2ae19a7af0c8de6e0528f935c10eb7c20a00081f1e3f6d5f6c
3
+ size 190481943
Faster RCNN/tortoise_label_map.pbtxt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ item {
2
+ name: "tortoise",
3
+ id: 1,
4
+ display_name: "tortoise"
5
+ }