update

Author: WidgetA

.gitignore

@@ -128,4 +128,9 @@ dmypy.json
 *.txt
 /object_detection/test_ckpt/
 /object_detection/test_data/
-/object_detection/test_images/
+/object_detection/test_images/
+*.meta
+*.index
+model_file/pretrain/checkpoint
+*.pb
+*.data-00000-of-00001

README.md

@@ -11,4 +11,5 @@
 - Tensorflow 1.15.2
 - lxml 4.5.0
 - Pillow 7.0.0
-- Matplotlib 3.2.0
+- matplotlib 3.2.0
+- PyYaml 5.3

model_train.py

@@ -0,0 +1,21 @@
+import os
+import shutil
+import argparse
+import sys
+
+parser = argparse.ArgumentParser()
+parser.add_argument("-path", help="Specify the absolute path to the compressed file")
+args = parser.parse_args()
+
+if not hasattr(args, 'path'):
+    print("Please specify the absolute path with \"-path\"")
+else:
+    os.chdir(sys.path[0])
+
+    if len(os.listdir('model_file/pretrain')) > 1:
+        pass
+    else:
+        shutil.copy(args.path, 'model_file/pretrain')
+    os.system('pip install -e slim')
+    os.system('python object_detection/legacy/train.py --train_dir=model_file/train ' +
+              '--pipeline_config_path=ssd_resnet50_v1_fpn.config')

object_detection/legacy/train.py

@@ -91,7 +91,8 @@
 @contrib_framework.deprecated(None, 'Use object_detection/model_main.py.')
 def main(_):
   assert FLAGS.train_dir, '`train_dir` is missing.'
-  if FLAGS.task == 0: tf.gfile.MakeDirs(FLAGS.train_dir)
+  if FLAGS.task == 0:
+      tf.gfile.MakeDirs(FLAGS.train_dir)
   if FLAGS.pipeline_config_path:
     configs = config_util.get_configs_from_pipeline_file(
         FLAGS.pipeline_config_path)

ssd_resnet50_v1_fpn.config

@@ -127,13 +127,13 @@ model {
 }
 
 train_config: {
-  fine_tune_checkpoint:"/Users/widget/Documents/Projects.nosync/Python.nosyc/sonar_baseline_with_tensorflow/model_file/pretrain/model.ckpt"
+  fine_tune_checkpoint:"model_file/pretrain/model.ckpt"
 from_detection_checkpoint: true
   batch_size: 16
   sync_replicas: true
   startup_delay_steps: 0
   replicas_to_aggregate: 8
-  num_steps: 25000
+  num_steps: 50000
   data_augmentation_options {
     random_horizontal_flip {
     }
@@ -168,9 +168,9 @@ from_detection_checkpoint: true
 
 train_input_reader: {
   tf_record_input_reader {
-    input_path:"/Users/widget/Documents/Projects.nosync/Python.nosyc/sonar_baseline_with_tensorflow/dataset/tfrecords/pascal_train.record"
+    input_path:"dataset/tfrecords/pascal_train.record"
   }
-  label_map_path:"/Users/widget/Documents/Projects.nosync/Python.nosyc/sonar_baseline_with_tensorflow/dataset/pascal_label_map.pbtxt"
+  label_map_path:"dataset/pascal_label_map.pbtxt"
 }
 
 eval_config: {
@@ -181,9 +181,9 @@ eval_config: {
 
 eval_input_reader: {
   tf_record_input_reader {
-    input_path:"/Users/widget/Documents/Projects.nosync/Python.nosyc/sonar_baseline_with_tensorflow/dataset/tfrecords/pascal_val.record"
+    input_path:"dataset/tfrecords/pascal_val.record"
   }
-  label_map_path:"/Users/widget/Documents/Projects.nosync/Python.nosyc/sonar_baseline_with_tensorflow/dataset/pascal_label_map.pbtxt"
+  label_map_path:"dataset/pascal_label_map.pbtxt"
   shuffle: false
   num_readers: 1
 }