Update

data/io/convert_data_to_tfrecord.py

@@ -8,7 +8,7 @@
 import tensorflow as tf
 import glob
 import cv2
-from libs.label_name_dict.remote_sensing_dict import *
+from libs.label_name_dict.label_dict import *
 from help_utils.tools import *
 
 tf.app.flags.DEFINE_string('VOC_dir', None, 'Voc dir')

help_utils/help_utils.py

@@ -3,7 +3,7 @@
 import tensorflow as tf
 import numpy as np
 import cv2
-from libs.label_name_dict.remote_sensing_dict import LABEl_NAME_MAP
+from libs.label_name_dict.label_dict import LABEl_NAME_MAP
 
 
 def show_boxes_in_img(img, boxes_and_label):

libs/box_utils/show_box_in_tensor.py

@@ -7,7 +7,7 @@
 import tensorflow as tf
 import numpy as np
 import cv2
-from libs.label_name_dict.remote_sensing_dict import LABEl_NAME_MAP
+from libs.label_name_dict.label_dict import LABEl_NAME_MAP
 
 
 def draw_box_in_img_batch(img_batch, boxes):

libs/label_name_dict/label_dict.py

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+from __future__ import division, print_function, absolute_import
+
+from libs.configs import cfgs
+
+if cfgs.DATASET_NAME == 'ship':
+    NAME_LABEL_MAP = {
+        'back_ground': 0,
+        'ship': 1
+    }
+elif cfgs.DATASET_NAME == 'aeroplane':
+    NAME_LABEL_MAP = {
+        'back_ground': 0,
+        'aeroplane': 1
+    }
+elif cfgs.DATASET_NAME == 'pascal':
+    NAME_LABEL_MAP = {
+        'back_ground': 0,
+        'aeroplane': 1,
+        'bicycle': 2,
+        'bird': 3,
+        'boat': 4,
+        'bottle': 5,
+        'bus': 6,
+        'car': 7,
+        'cat': 8,
+        'chair': 9,
+        'cow': 10,
+        'diningtable': 11,
+        'dog': 12,
+        'horse': 13,
+        'motorbike': 14,
+        'person': 15,
+        'pottedplant': 16,
+        'sheep': 17,
+        'sofa': 18,
+        'train': 19,
+        'tvmonitor': 20
+    }
+else:
+    assert 'please set label dict!'
+
+
+def get_label_name_map():
+    reverse_dict = {}
+    for name, label in NAME_LABEL_MAP.items():
+        reverse_dict[label] = name
+    return reverse_dict
+
+LABEl_NAME_MAP = get_label_name_map()

tools/eval.py

@@ -10,17 +10,15 @@
 import os
 import time
 from data.io.read_tfrecord import next_batch
-from libs.configs import cfgs
 from libs.networks.network_factory import get_network_byname
-from libs.label_name_dict.remote_sensing_dict import *
+from libs.label_name_dict.label_dict import *
 from libs.rpn import build_rpn
 from help_utils.tools import view_bar
 from tools import restore_model
 import pickle
 from libs.box_utils.coordinate_convert import *
 from libs.box_utils.boxes_utils import get_horizen_minAreaRectangle
 from libs.fast_rcnn import build_fast_rcnn
-from libs.box_utils import iou_rotate
 
 os.environ["CUDA_VISIBLE_DEVICES"] = '1'
 

tools/eval1.py

@@ -10,9 +10,8 @@
 import os
 import time
 from data.io.read_tfrecord import next_batch
-from libs.configs import cfgs
 from libs.networks.network_factory import get_network_byname
-from libs.label_name_dict.remote_sensing_dict import *
+from libs.label_name_dict.label_dict import *
 from libs.rpn import build_rpn
 from help_utils.tools import view_bar
 from tools import restore_model