Change dtype of im_id to int64 (#148)

ppdet/modeling/architectures/blazeface.py

@@ -176,7 +176,7 @@ class BlazeFace(object):
         # yapf: disable
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
             'im_shape': {'shape': [None, 3], 'dtype': 'int32',   'lod_level': 0},

ppdet/modeling/architectures/cascade_mask_rcnn.py

@@ -371,7 +371,7 @@ class CascadeMaskRCNN(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},

ppdet/modeling/architectures/cascade_rcnn.py

@@ -283,7 +283,7 @@ class CascadeRCNN(object):
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
             'is_crowd': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},

ppdet/modeling/architectures/cascade_rcnn_cls_aware.py

@@ -176,7 +176,7 @@ class CascadeRCNNClsAware(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},

ppdet/modeling/architectures/faceboxes.py

@@ -150,7 +150,7 @@ class FaceBoxes(object):
         # yapf: disable
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
             'im_shape': {'shape': [None, 3], 'dtype': 'int32',   'lod_level': 0},

ppdet/modeling/architectures/faster_rcnn.py

@@ -194,7 +194,7 @@ class FasterRCNN(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},

ppdet/modeling/architectures/mask_rcnn.py

@@ -274,7 +274,7 @@ class MaskRCNN(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},

ppdet/modeling/architectures/retinanet.py

@@ -88,7 +88,7 @@ class RetinaNet(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},

ppdet/modeling/architectures/ssd.py

@@ -95,7 +95,7 @@ class SSD(object):
         # yapf: disable
         inputs_def = {
             'image':        {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
-            'im_id':        {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':        {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':      {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class':     {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
             'im_shape':     {'shape': [None, 3], 'dtype': 'int32',   'lod_level': 0},

ppdet/modeling/architectures/yolov3.py

@@ -96,7 +96,7 @@ class YOLOv3(object):
         inputs_def = {
             'image':    {'shape': im_shape,                 'dtype': 'float32', 'lod_level': 0},
             'im_size':  {'shape': [None, 2],                'dtype': 'int32',   'lod_level': 0},
-            'im_id':    {'shape': [None, 1],                'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1],                'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':  {'shape': [None, num_max_boxes, 4], 'dtype': 'float32', 'lod_level': 0},
             'gt_class': {'shape': [None, num_max_boxes],    'dtype': 'int32',   'lod_level': 0},
             'gt_score': {'shape': [None, num_max_boxes],    'dtype': 'float32', 'lod_level': 0},