Fix onnx export (#239)

classification/export.py

@@ -19,7 +19,7 @@ def get_args():
     parser.add_argument('--model_name', type=str,
                         default='internimage_t_1k_224')
     parser.add_argument('--ckpt_dir', type=str,
-                        default='/mnt/petrelfs/share_data/huangzhenhang/code/internimage/checkpoint_dir/new/cls')
+                        default='pretrained/')
     parser.add_argument('--onnx', default=False, action='store_true')
     parser.add_argument('--trt', default=False, action='store_true')
 

classification/ops_dcnv3/functions/dcnv3_func.py

@@ -100,7 +100,7 @@ class DCNv3Function(Function):
             group_channels_i=int(group_channels),
             offset_scale_f=float(offset_scale),
             im2col_step_i=int(im2col_step),
-            remove_center=int(remove_center),
+            remove_center_i=int(remove_center),
         )