search

facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
30.7k stars 7.51k forks source link

export FPN-based Rotated R-CNN to onnx #1955

Open freedenS opened 4 years ago

freedenS commented 4 years ago

Instructions To Reproduce the 🐛 Bug:

  1. what changes you made (git diff) or what code you wrote 
    None
  2. what exact command you run: python tools/deploy/caffe2_converter.py --format onnx --config-file model/testConfig.yaml --output ./output/onnx_model --run-eval MODEL.WEIGHTS ./model/model_final.pth MODEL.DEVICE cpu(cuda)
  3. what you observed (including full logs): 
    [08/25 13:48:28 d2.data.common]: Serializing 100 elements to byte tensors and concatenating them all ...
[08/25 13:48:28 d2.data.common]: Serialized dataset takes 0.46 MiB
[08/25 13:48:28 d2.data.dataset_mapper]: Augmentations used in training: [ResizeShortestEdge(short_edge_length=(800, 800), max_size=1333, sample_style='choice')]
/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/onnx/utils.py:243: UserWarning: `add_node_names' can be set to True only when 'operator_export_type' is `ONNX`. Since 'operator_export_type' is not set to 'ONNX', `add_node_names` argument will be ignored.
"`{}` argument will be ignored.".format(arg_name, arg_name))
/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/onnx/utils.py:243: UserWarning: `do_constant_folding' can be set to True only when 'operator_export_type' is `ONNX`. Since 'operator_export_type' is not set to 'ONNX', `do_constant_folding` argument will be ignored.
"`{}` argument will be ignored.".format(arg_name, arg_name))
/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/detectron2/export/c10.py:31: TracerWarning: Converting a tensor to a Python boolean might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future. This means that the trace might not generalize to other inputs!
assert tensor.dim() == 2 and tensor.size(-1) in [4, 5, 6], tensor.size()
Traceback (most recent call last):
File "tools/deploy/caffe2_converter.py", line 70, in <module>
onnx_model = tracer.export_onnx()
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/detectron2/export/api.py", line 127, in export_onnx
return export_onnx_model_impl(model, (inputs,))
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/detectron2/export/caffe2_export.py", line 60, in export_onnx_model
operator_export_type=OperatorExportTypes.ONNX_ATEN_FALLBACK,
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/onnx/__init__.py", line 208, in export
custom_opsets, enable_onnx_checker, use_external_data_format)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/onnx/utils.py", line 92, in export
use_external_data_format=use_external_data_format)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/onnx/utils.py", line 530, in _export
fixed_batch_size=fixed_batch_size)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/onnx/utils.py", line 366, in _model_to_graph
graph, torch_out = _trace_and_get_graph_from_model(model, args)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/onnx/utils.py", line 319, in _trace_and_get_graph_from_model
torch.jit._get_trace_graph(model, args, strict=False, _force_outplace=False, _return_inputs_states=True)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/jit/__init__.py", line 338, in _get_trace_graph
outs = ONNXTracedModule(f, strict, _force_outplace, return_inputs, _return_inputs_states)(*args, **kwargs)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/jit/__init__.py", line 426, in forward
self._force_outplace,
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/jit/__init__.py", line 412, in wrapper
outs.append(self.inner(*trace_inputs))
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/nn/modules/module.py", line 720, in _call_impl
result = self._slow_forward(*input, **kwargs)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/nn/modules/module.py", line 704, in _slow_forward
result = self.forward(*input, **kwargs)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/contextlib.py", line 52, in inner
return func(*args, **kwds)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/detectron2/export/caffe2_modeling.py", line 272, in forward
detector_results, _ = self._wrapped_model.roi_heads(images, features, proposals)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/nn/modules/module.py", line 720, in _call_impl
result = self._slow_forward(*input, **kwargs)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/nn/modules/module.py", line 704, in _slow_forward
result = self.forward(*input, **kwargs)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/detectron2/modeling/roi_heads/roi_heads.py", line 674, in forward
pred_instances = self._forward_box(features, proposals)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/detectron2/modeling/roi_heads/roi_heads.py", line 727, in _forward_box
box_features = self.box_pooler(features, [x.proposal_boxes for x in proposals])
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/nn/modules/module.py", line 720, in _call_impl
result = self._slow_forward(*input, **kwargs)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch/nn/modules/module.py", line 704, in _slow_forward
result = self.forward(*input, **kwargs)
File "/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/detectron2/export/c10.py", line 312, in forward
legacy_plus_one=False,
RuntimeError: [enforce fail at collect_and_distribute_fpn_rpn_proposals_op.cc:17] (areas >= 0).all(). Negative areas founds:    96.2632    141.673    67.3605    52.5703
     0    98.9163     146.05    75.7181
3.21926          0    100.779    146.512
286.707    38.4453    208.841    291.735
57.4213   -18.7202          0    93.2274
     0    490.569    65.6323     273.69
65.7779    52.7958    19.1413          0
0.339808          0    489.396   -24.0795
309.316   -60.3876          0    492.672
332.346    195.418   -1.75527          0
-14.8156    210.573    304.644    56.4441
272.997    25.5623    215.736    282.077
     0    456.934    6.70247    269.389
132.734    61.4397    53.5414    19.8702
108.014    138.886    80.1393    46.6009
-1.73336          0    254.128    65.8985
131.797  -0.971659          0    265.267
9.65313   -17.4754    434.908    619.688
229.177    289.866   -46.6854          0
34.1459      282.4    212.094    1.05489
     0      111.3    146.146    71.5676
508.421    4.61457    270.951    254.333
     0    445.041    -14.277    211.789
56.7109          0    297.425   -8.75534
295.763   -50.4786          0    5.27159
     0    345.608   -12.7634    436.138
-48.3036          0    576.808   -32.1141
3.87991          0    110.318    139.511
221.952    315.073   -58.3937          0
596.652   -54.2146          0    29.5274
490.472    501.673    65.8184          0
-3.75398    269.921    264.666   -1.92122
550.453   -23.3852    215.323    303.658
     0    445.947    19.7731    227.435
57.4134   -19.5825          0     98.936
36.798          0    255.497   -28.0422
298.824     40.165          0    320.349
217.764    302.089   -39.9313          0
84.2966    42.8887   -23.9179          0
52.1842    332.274    104.368   0.418602
249.968    26.6464     197.02    284.327
     0    267.538  -0.203382    336.021
-7.92594    683.863    379.607    3.55073
129.419    144.134    258.839    288.267
2.59747          0    546.306   -6.76236
     0    610.919   -9.21827     538.52
304.309    52.8299          0    3.16197
-1.63364          0   0.307782    -58.812
214.662     315.03    59.1647          0
5.87843    211.191    296.848   -34.1561
481.642   -24.9236          0    576.765
461.047    583.376    50.3331          0
559.903   -2.65931    276.168    252.952
40.1294    532.498     501.65    54.4901
538.688    24.9176    571.399    485.421
148.543    65.7405    53.6867    24.9813
     0     552.72   -3.59984    343.628
-1.73987          0    291.448    61.7823
     0     430.67    11.0978    713.772
246.044    22.8712          0    435.657
-1.56322          0    594.692    6.33347
323.299    198.998     25.537          0
390.466   -4.18124          0 -0.0816266
-0.216221    251.545    244.231    63.9729
410.155  -0.638356    267.808    244.159
 547.4    500.632   -67.3531          0
     0    243.122    53.6453    253.205
-3.37365          0    218.641     2.0612
8.13857    599.786    550.102   -2.16589
334.177    132.552    545.122    265.104
299.605    -5.2178          0    317.537
225.765    296.684    47.8258          0
-2.91741    343.597     205.34   -2.60734
-9.53062    2.02079    355.095    191.491
     0   -31.3843   -2.26852    689.981
-29.6125          0    616.844    36.0387
528.313   -59.2305          0    390.341
     0    306.337    31.5448    215.002
37.8717          0     6.0974    22.1471
251.152   -69.0552          0    581.429
208.857    304.526    -56.842          0
101.006    159.479    61.5311     49.623
12.6175    286.402    297.987   -4.32291
     0    296.943    187.953    77.7674
364.88    1.43835    279.396    271.179
     0    409.232     14.703    231.695
32.166          0    352.298   -22.3481
300.726   -53.7527          0     306.75
336.507    180.293   -21.4803          0
476.249    657.249    45.6844          0
2.4526    348.408    196.265   -3.15575
5.92671          0    177.503    139.473
31.4062    -13.322    357.648    164.857
     0    376.068   -14.5281     220.61
48.1607          0    60.2243   -14.3777
242.01   -29.2177          0    235.503
40.467    15.2315          0    84.8578
217.946    273.301    23.2085          0
65.3786    52.6832    9.04277          0
137.981    83.9727    56.3302 -0.0356861
32.5499    539.088    467.038    50.9514
171.999    131.141    67.0121    40.3553
236.009    11.0773    535.854    575.125
     0    358.061    107.437    650.978
27.3979     204.96    299.589    -52.334
325.527     41.812    301.846    206.425
     0    508.264    42.1075    274.388
-8.20999          0    306.702    11.8321
173.615   -30.7453          0    273.055
288.896    260.584   -28.7712          0
-0.756652          0    464.055   -9.60313
36.2662    208.148    292.004   -49.9789
256.815    4.25627     361.77    178.461
     0    50.5105    206.266    76.4367
     0    204.564     18.491    278.743
46.9053          0     464.33    38.4735
   200   -28.3291          0    220.434
-1.59555          0    73.0992     175.54
626.112   -58.2782          0    580.777
343.214    101.779   0.269551          0
53.9058    9.70611          0    112.445
716.825    369.708   -28.6889          0
-25.1628    449.979    586.767    39.8105
7.6195    294.054    259.323   -58.2011
642.153   -20.8137    418.732    577.459
494.794    37.3178    317.571    200.471
     0     33.472    64.9491    247.089
29.6635          0    608.808    3.82392
     0    188.914    124.265    377.828
193.888   -1.74614          0    347.666
343.706    172.624   -30.1559          0
-6.19568    213.168    283.051   -50.4532
369.123    10.1995    262.455    226.917
     0    626.652    69.9287    279.417
-0.908602          0   -7.35107    330.676
313.307    51.8405          0     212.03
0.380503          0    182.506   -38.5353
267.223     248.48   -5.62748          0
77.2577    42.7971   -0.39423          0
12.0203    221.607    290.558    28.4337
575.583    37.2611          0     492.58
452.427     556.16    32.4154          0
145.431      78.65    56.4081   -16.4264
186.521    218.477    213.778    293.244
     0    661.761   -4.99074    281.484
-2.80199          0    17.0398    28.6038
205.29    2.01148          0    245.898
179.042    133.249    64.6029     40.557
346.607    179.611    30.0349          0
35.9009    530.972    483.403    44.2347
33.8621    354.409    196.437   -26.6257
638.775  -0.763083    230.866    294.833
     0    244.829    165.504    263.858
-58.845          0    512.719    60.8378
140.938     11.719    584.637    558.461
278.446   -57.0377          0  -0.779353
     0    18.2805    131.285    530.403
276.856    239.309    1.42163          0
-26.7742          0    179.942    -10.118
185.252    268.684    219.364     51.719
214.182     207.47    253.472    238.206
     0    427.625    33.3135    273.704
45.5159          0    24.9598    371.694
393.541    1.33459          0    71.9337
505.029    490.984   -31.7779          0
2.48504    555.152    492.893   -27.4774
     0    105.792    122.209     66.714
-50.7378    71.2622    542.727    470.334
     0    306.929    43.0476    675.748
267.644   -52.0456          0    467.792
11.6759          0        309    191.655
51.3059   -2.37862          0    92.3953
75.3344    52.7184    1.17125          0
240.767    294.796    30.1946          0
143.775    206.998    304.118    48.9945
13.275    17.1387    203.151    393.874
     0    190.727    200.327    279.264
-3.96754          0    588.216    10.7223
174.442   -28.1385          0    304.551
-29.4633          0     326.01    30.6423
360.441    184.481   -24.2299          0
569.452    22.5692          0    74.0879
-3.72528    160.649    130.727     116.63
101.343    248.961    286.139    24.1154
13.3388    368.593     214.23    306.687
     0    182.511    158.945    212.881
53.4612          0     345.93    58.8598
176.297   -26.4439          0    188.431
218.769    306.633    52.0678          0
32.9554    323.532    257.753   -18.8643
143.432    174.428     266.58    245.487
691.986     364.88    -32.704          0
     0    236.414    19.3813     279.98
17.0078          0    172.592    144.072
237.091    56.0762          0   -8.99861
207.797    306.179    56.4735          0
-48.7182    668.777    379.506    22.0728
-16.8417    189.173    464.969    640.922
129.49    135.527    186.091   0.618134
     0    268.811   -62.9413    650.528
26.9843          0    11.6825    151.695
144.486    70.7655    54.8339   -13.4414
480.228    13.1023    200.874    373.499
     0    607.667    11.1601     233.89
539.559   -55.6181          0    150.141
449.422    583.156    21.9811          0
-27.985          0    365.425    75.5916
299.408    415.326    649.936    56.9907
224.092    38.8066    658.968     337.76
151.183  -0.714459          0     640.73
228.236     279.02    23.6267          0
48.3356    226.009    96.6712   0.749593
105.879     147.54    277.306    241.352
     0   -46.5455    205.738    764.268
-2.1618          0   -2.44627    217.331
644.892   -64.1627          0    238.955
     0    500.954     39.644    221.086
 27.79          0    184.712    232.793
180.35    136.072    79.1125    47.3094
188.143   -29.8562          0     370.83
302.591    278.481    -6.4732          0
206.86    221.637    284.333    16.9948
-1.1214    168.469    283.988    255.999
     0   -15.5081    380.814    262.056
     0    99.9839    129.597    79.7237
56.3407          0   -12.1163    141.047
245.632   -4.03268          0    227.511
-0.533768          0    295.304    197.411
313.382    247.605   -27.3153          0
544.022    512.543   -59.7121          0
133.143    719.868     367.44    1.26513
159.099    227.459    290.616    24.3702
447.575   -11.2957    333.223    176.808
     0    7.15046    364.484    264.629
-32.6795          0    44.2616    62.5744
45.9226    5.13312    701.781    327.243
247.828   -24.0901          0    204.236
     0   -7.95517    138.698    105.195
348.042    167.638   -30.7753          0
159.696    339.649     201.13  -0.281063
     0    18.3499    129.284    437.134
148.916    144.928    289.885    237.511
     0     686.75    -3.4802    252.476
28.8852          0    -1.2947   -30.2151
-58.3055          0    44.2665    355.022
60.0702    11.4435          0    75.1572
339.78    3.37489          0    55.0107
272.489    271.342   -56.1312          0
168.926    158.369    192.971  -0.706482
75.9707    55.1903   -15.6516          0
295.183    109.808    281.617    254.668
     0     239.11    6.44633    274.253
-23.6862          0    178.801   -22.3688
546.275   -56.0309          0    107.883
57.9289          0   -5.57461    164.386
301.333    29.5554          0  -0.847825
211.298    302.863   -58.0447          0
112.467    29.6819          0    4.19057
416.217    307.646    277.685   -55.2218
677.795     2.9752    344.762    190.671
438.555     580.37    35.9799          0
-7.50602    384.974    786.237    28.4392
     0    454.136    29.5457    320.364
142.751     127.04   -1.01098          0
23.5134          0    70.9755    400.634
245.87  -0.817474          0     459.11
354.52    167.684   -93.9707          0
136.436    58.3139    46.1546    13.3937
144.427    317.159    189.693    25.5112
399.954    29.4612    335.675    230.879
2.5479    364.616     560.74    512.158
     0    197.323    208.177    343.557
-1.11109          0    -4.0751    179.809
     0    144.466   -49.5813    665.826
314.411   -35.9537          0    672.307
26.5708          0    85.9848    202.137
144.471    140.302    110.734    30.3599
336.652    197.068   -1.90966          0
102.201     131.73    66.5824    50.3446
19.9385    198.505    372.915    23.8446
20.5521    31.0829    331.824    196.637
     0   -2.56291    217.862    290.987
     0    209.356    222.715    70.9111
-27.4301          0    488.011    14.0612
4.30595          0    83.8083    182.774
-5.86418   -11.8591     127.42    116.452
560.924    21.1212          0      282.6
178.993     63.304          0   -4.78873
744.413    355.714   -30.6481          0
62.3432    -10.708          0    89.1246
200.494    330.513    37.0886          0
162.343    353.802    181.725   -29.3265
203.13    596.966    476.977   -8.14528
69.5011    172.113    139.002    249.672
87.7695    43.2068   -26.3325          0
     0    417.228   -6.42728    346.667
-2.8215   -72.2272    332.819    690.836
-92.0972          0    98.5269   -5.04224
300.412   -42.7622          0    168.415
274.578    242.752    44.2395          0
135.254    82.0053    44.2855    25.3357
     0   -95.6721    14.9269    659.106
142.826    228.399    312.194   -30.5708
-1.79397    158.906    211.498    330.865
87.2308          0    243.641    79.9318
     0    128.221    172.064    215.349
85.6395    139.001    72.3986    42.8671
576.366    24.5216          0   -39.3192
332.386     751.47   -29.3756          0
-50.775          0    194.862    258.538
243.73   -24.3908          0     259.64
254.508    241.808   -21.6501          0
374.599    322.776    191.527   -27.8739
-45.3783     7.4344     360.62    177.784
376.768     401.69    638.284   -55.9146
     0    1.47414    98.7822    259.077
     0    59.8189    210.017    77.4076
-34.3531          0     234.27    172.215
152.517    181.441    532.157    461.324
     0    169.524    105.525    506.262
173.871   -30.3014          0    194.518
346.843     178.74    27.6707          0
14.8838          0    80.1805    169.331
182.754    339.123    188.014    24.3523
-9.70499    2.03419    347.137    184.637
     0    19.7724    160.072    145.396
2.87816          0    206.892    220.422
42.7857   -12.4962          0    112.236
477.147    51.4095          0    84.5454
     0      173.7   -2.72163    343.432
-1.70929          0    1.79762    235.309
324.203   -63.2479          0    60.7401
333.862    184.248    -27.436          0
238.544    285.144    247.096   -32.5002
77.2005    42.2471   -29.1162          0
33.0902    328.906    258.462    220.792
     0   -8.59031    198.835    337.484
-57.577          0   -7.01809      154.7
29.9405          0    -28.452    336.463
685.893    412.162   -5.31188          0
202.779    21.8842          0    190.799
136.88    61.6552    48.5218    20.6721
214.815     303.59   -56.2864          0
215.027    245.837    73.7638     45.011
     0    54.1653    217.826     86.259
 333.8    283.788     252.62   -1.14335
99.6342   -25.7587          0    36.5169
170.528    420.581    598.496    50.4058
175.183    160.057    210.894    271.106
5.80913          0    296.589    192.162
49.8205   -28.3987          0     194.93
80.9787    43.2339   -18.8906          0
     0    148.743    235.451    356.018
138.35    62.0437     52.745   -21.2239
-29.9389          0    409.322    50.3589
287.574   -34.3821          0    240.361
259.524    259.543    22.0084          0
263.474    218.841    282.702    21.2049
434.043    79.2706    258.192    281.778
296.962    189.824    93.9794    51.8367
     0    137.699     179.71    342.554
-32.4914          0     264.55    95.9051
180.654   -27.2114          0    519.433
350.677    175.151    63.6565          0
48.8516    276.429    264.624      34.79
     0     163.57    138.918    62.0549
7.99265          0    50.6992    213.732
227.018    275.835    218.482    305.078
     0    249.407    235.299    272.026
21.6755          0      84.56    -27.393
303.431    35.2091          0    240.689
-23.2434   -23.3094    613.782    357.361
282.579    252.292   -5.32496          0
43.6812   -4.99682          0    125.371
-9.42306    177.446    343.025   -25.9549
73.0338    94.4939   0.535755          0
-5.49226    365.425    328.731    289.461
     0   0.920289     316.98    349.245
     0   0.621827    236.169    490.977
1.7481          0    145.783        165
83.0154    53.4739    5.37882          0
139.466    103.647    151.915    27.7661
-28.3653          0    196.519    229.344
195.176   0.962513          0    730.509
281.086    267.266   -5.56912          0
3.41817    340.003    189.236   -30.2787
-13.8122    245.713     210.63    305.969
238.807    78.9467    45.3382   0.536596
     0    380.102   -2.95891    341.554
82.0344          0   -4.70394    201.038
95.7927    153.403    62.2106    55.0675
     0    200.178    206.618    79.9473
192.369    1.06419          0    77.8943
302.528    236.466   -42.9741          0
-5.84357    350.747    166.488   -95.6011
421.859    23.7253    323.066    174.369
     0   -7.26805    198.768     367.07
-28.2706          0    208.058    38.6561
386.133   -29.4835          0     87.317
1.68845     173.85    176.024    88.4667
174.756    23.7676          0    86.4636
337.973    184.489    29.2936          0
-6.67743    320.228    168.792    26.0912
219.355    269.943    327.529    188.421
     0    229.826    199.254    261.353
-21.9455          0    56.8946    163.896
253.086   -55.7394          0    195.902
339.43    198.344    0.23482          0
59.5056    258.161    292.359    -4.9627
151.827    303.949    224.886    283.498
18.0401          0     303.24     185.21
52.4135    27.0658          0    294.782
     0    14.4244    177.168    198.838
     0  -0.597441    178.731        136
108.41    53.9806    22.9554          0
26.2211          0    92.6968    27.6275
274.864   -25.5007          0    150.967
224.724    293.277    24.7534          0
217.068    242.757    320.791   -25.9059
127.52    146.311    331.458    181.406
     0    172.055    241.367    266.437
23.0056          0    238.618    55.6175
258.778    35.1164          0    -18.502
327.185    183.323   -66.1454          0
561.522    514.266    23.1023          0
52.6507    286.276    262.786   -7.78687
223.018    196.753    204.653    322.513
     0   -25.7526    188.493    378.153
130.667    60.8399    42.9277    7.49299
-92.2022          0     117.42    271.665
309.42     192.79    94.9145    58.6005
     0    173.745    131.292    75.8879
1.77906          0    111.623    152.786
-26.6432          0   -11.7088    165.383
175.612   -27.0797          0    557.823
204.226    378.694    22.4185          0
87.6093    248.158    273.547   -31.6244
203.734    112.081    264.304    240.097
49.0578   -13.1206          0    229.247
     0    718.025    14.0276    291.086
71.7133    52.1864   -4.43355          0
139.661    72.2756    49.1168   -14.4665
215.783    403.366    607.007   -55.9607
122.207    231.652    717.722    385.434
-53.1345          0    445.694    55.3213
     0     45.347    330.501    643.593
156.926    21.4456          0   -14.3383
3.16561          0    256.654    120.916
292.108   -7.54836          0  0.0947728
79.1481    184.227     65.994     51.776
297.291    246.593   -27.7692          0
392.267    354.961    225.325    1.81589
464.74   -28.6945          0    14.5575
5.20861    235.228    356.093    177.205
88.2519    174.916    2.56009          0
     0    54.6082    39.2571    209.237
25.2512          0    77.4301    134.922
187.238   -1.96504          0    553.859
532.81    505.636   -5.04424          0
346.745    183.057   -63.1999          0
5.6783     176.88     97.193   -22.7892
-18.4147    2.46807      167.5    101.913
329.143    506.658    696.324   -35.7908
129.46     274.92    247.477    -1.6878
198.638    131.212    372.138    186.727
     0    124.026    231.044    283.923
0.513961    208.597    38.4166    29.8454
-6.42913          0    420.927    71.9039
-34.7277    388.609    441.192    573.829
143.808  -0.819074          0    255.514
     0     129.13    15.8458    681.395
286.621    224.046    18.9639          0
138.557     300.34    215.627     23.485
283.311    169.932    259.959    263.423
-26.1423          0   -37.1404    313.509
     0    90.7559    227.156    181.512
-0.44065          0    561.304    28.3654
593.493    32.0559          0   -46.5901
176.493    61.7918          0    85.2601
284.511    244.999    1.50173          0
145.312    359.077    152.694   -95.7796
213.59    77.4527    260.453    262.238
     0    206.507    169.073     69.661
639.97    391.832    26.7653          0
270.245    641.103    394.388    1.21531
86.9638   -32.9404    542.184    465.597
     0    121.081    128.772    292.639
 4.855          0    236.661    133.725
     0    149.492    294.497    447.394
-50.215          0    230.245    194.928
389.856  -0.580003          0    46.0247
-28.9745          0    408.131    27.5453
44.1048    2.82595          0    62.2474
65.4823    52.4336    5.03628          0
371.513    24.8054          0    114.591
240.106     293.32    45.3146          0
232.912    331.939    192.898   -1.07847
196.76     112.39    315.343    190.959
     0    139.469   -22.5381    262.893
26.0386          0   -17.8676    281.239
299.646    29.9288          0     -15.61
     0   0.809489    -1.2992    138.195
193.396    88.5973    48.1135   -27.7739
198.441    357.631    4.38306          0
-14.4614    173.431    365.418    3.35977
103.672    66.2621    207.343    132.524
120.539    134.794    77.8753    50.6576
     0    364.244    28.3693    354.369
-27.0005          0    299.143    29.9118
361.443    26.3855          0   -38.5957
166.297    384.749   -24.9099          0
     0    51.5188    228.349    80.5423
-54.2145    151.904     346.86   -1.28045
598.686    512.838   -4.60264          0
8.16104          0    2.98714    224.509
192.927    8.08128    277.205    224.409
55.1397   -33.3921          0    215.179
     0    156.213    7.56079    319.478
161.203    371.638    765.689     26.576
110.555    309.403    447.476     499.83
58.1675    44.7446    7.96714          0
-7.08525    61.1456    65.2348   -42.3651
     0    515.285    44.8018    650.453
61.6889          0    549.659    84.6446
776.807    26.6255          0    16.9736
232.413    146.275    75.2896    44.0974
685.393    390.249    1.32493          0
33.696          0   -14.0577    165.133
96.1075    1.55245          0   -7.26882
     0    3.47104    211.312    62.1683
-40.9455    685.928    364.962   -63.1481
-36.1931          0   -4.26947    212.122
58.1366    26.8229          0    88.4915
180.929    84.0194    33.7501          0
91.0899    42.6269   -27.0549          0
212.257    78.7273    43.2222    -16.363
17.6115    74.5654    674.366    392.821
84.2935    180.526    70.4399    44.4151
     0   -42.4559    55.1826    662.348
7.58797    155.818    93.6096    63.7732
     0   -4.36772    222.213      64.95
29.6205          0    248.691    134.908
50.4122    4.08226          0    167.487
94.1285    42.0436      22.67          0
168.118    145.604    139.753     111.95
-66.9475          0    175.321     148.12
432.41    25.9012          0    600.288
667.397     357.13    62.8005          0
87.8871    594.229    393.761   -2.04669
177.122    70.3455    46.1218  -0.542799
102.724     140.14    82.5906     42.575
618.323     57.608    405.351    796.993
     0   -73.7169    138.839    706.453
     0     221.97    254.371    67.3605
-92.6633          0   -30.6165    203.971
     0    223.102    155.526    144.753
-24.2864          0   -8.37904    5.82761
-3.55728          0    87.0695     140.81
146.401    8.07834          0    44.4895
42.2311    -26.412          0    192.571
     0   -6.01916    209.403    47.8096
772.175    5.04115          0   -67.8314
320.714    716.724   -27.0255          0
-24.9275    662.765    357.678    -62.523
60.6542    259.089    371.782    792.204
     0   -23.3749    176.569    697.089
-61.3868          0    139.466    269.341
88.979     49.179 -0.0411879          0
584.72   -19.3295          0     487.28
351.944    689.732   -26.3236          0
71.8881    49.5021    6.76656          0
157.146    541.703    568.838   -3.94097
128.07     75.227    35.2377   -7.85543
186.899    150.749    109.295    25.0793
77.3046    229.748    154.609    459.496
8.18978    6.52511     95.013    194.402
-1.03803    2.78503    74.0341    64.9057
     0    87.3028    127.099    67.5639
     0    202.333        157    146.506
11.3383          0    96.0138    132.903
     0   -81.1297    244.077     318.53
53.1746   -20.0894          0    265.942
-25.3498          0    50.5747     430.41
508.278   -45.6165          0     423.45
70.5411     54.377    18.0792          0
200.837    65.4535    54.0808    19.6118
617.344    373.247   -61.9195          0
131.789    450.127    563.679    38.7304
596.377   -22.0678    358.939    695.465
241.267    143.792    67.9445    40.8453
-26.6492          0    153.932    154.934
117.06   -27.9753          0   -19.7228
     0    84.3041     175.21    83.8278
26.2226          0    176.231    248.303
44.8666   -4.75911          0    90.7452
57.0818    47.5907    -5.1261          0
     0   -28.9107    109.664    613.587
89.3316          0   -74.6243    238.474
203.744    66.2775    54.5134   -24.3813
134.653    124.022    62.3579          0
17.378    15.7462     34.756    31.4925
     0    105.487    129.475    84.1326
332.087    92.3026          0   -1.21425
189.702    187.089    88.8786   -29.3895
13.7452          0   -3.05649    209.099
442.039    544.202    7.49652          0
43.7117   -1.57613          0    168.388
71.7005    37.3168   -5.06578          0
-30.4613    678.101    321.734   -93.6414
211.565    157.038    133.312     117.38
0.954975    82.0808    49.2933    1.59291
76.9148    253.115    450.521    608.922
211.293    211.687    80.1693    47.7375
     0    181.416     255.26    67.4905
-4.39097          0    172.529    130.754
     0      24.59   -132.353    303.628
5.54391          0    775.106   -24.8532
41.9898    29.7296          0    312.281
543.155    -2.0133          0    302.842
77.5279    54.8866    25.8866          0
-3.74514    67.7208    54.7957    16.1162
-3.8549    231.453    66.7681    51.8123
709.708    315.313   -97.4617          0
     0    210.216    174.253    147.101
189.092    556.068    472.516   -24.8554
     0    181.674    128.175    76.3583
30.9003          0   -22.6419    126.364
-51.9566    365.014    753.727    330.547
74.8657    89.8985          0    11.5798
16.4136          0    288.162    192.965
175.576    93.7488    61.1167          0
     0    380.525   -33.1342    711.602
47.0061   -28.0054          0    -6.0373
80.3238    42.4179   -2.33755          0
-93.2723          0    392.556    7.88558
209.768    127.975    119.775    -21.386
2.20135    89.8708    47.0096   -26.4705
696.993   -24.5769          0    213.117
576.23    498.922   -4.08061          0
-25.2805    669.652    317.533   -93.3061
716.688    160.026    602.379    555.786
8.15561    186.464    139.361    130.094
     0   -14.9171    151.136    82.1934
-28.802          0    130.448    134.747
106.547    55.8647          0      5.482
92.7672    123.337    75.2923    48.1449
     0     16.787    225.044     33.574
-0.857639          0    161.438    128.269
     0    395.221    81.7859    383.262
51.0521   -4.26764          0    221.741
65.3176    42.5384    6.33869          0
94.4092    206.015    23.6251          0
2.17031          0  -0.278076    37.9256
173.087    148.469    107.837    29.9459
196.468    156.028    171.005    93.5526
760.444    3.41705          0    51.5408
     0   -18.9493    131.207    142.995
25.052          0    165.922    157.367
5.6314      63.97    62.5815    15.5426
2.00039          0   -2.35207    210.708
195.835    218.396    77.7051    46.8693
144.602    -12.573          0    237.081
288.561    777.916   -26.9081          0
267.365    669.573    336.865    58.6646
721.856    110.911    657.379    221.823
142.781    115.671    22.0556          0
     0    3.44543     -2.919    72.7043
151.49    127.667    114.455    -19.799
203.937    164.505     191.64    96.7865
     0    146.935    153.368    187.072
19.1833          0    67.6042    180.692
     0    73.4454    250.062     555.46
4.89802          0    -2.3012    220.407
48.0274          0    382.109    90.6304
48.0475   0.604202          0    120.768
65.7911    44.8638    8.97778          0
710.341    28.0132          0    107.938
199.454    31.3916    44.7328   0.748803
253.042   -7.83778    69.7078    56.5312
     0    126.791    140.554    76.5459
25.6254          0    73.1947    212.269
639.96    352.816   -30.2889          0
64.6959    16.2985          0    80.1333
105.05    23.3837          0    144.336
76.2012    40.7283     5.7027          0
172.987    78.1129    54.4295    4.51991
84.866    165.209    63.3273    54.0225
     0    229.474    127.005    59.6481
7.16221          0    270.412   -6.46271
329.809    642.017    327.671    53.4863
54.6864   -24.3882          0    126.247
79.9149    46.2351    26.2194          0
151.825     72.761    42.6335    5.55255
131.961    121.594    2.05634          0
660.872   -25.3862    659.082    360.451
223.707     241.39    71.7028    46.7745
172.196    81.5346    178.804   -24.6857
93.5044    145.427    130.994     113.01
     0    170.639    140.839    88.1404
     0    649.208    71.2879    441.978
-26.2626          0    214.646    234.836
-37.5769          0    654.266    114.293
378.852   -29.7523          0    273.301
546.602    394.134   0.323527          0
48.1574   -1.98111          0    101.531
29.0909     676.21    374.444    -31.022
698.246    77.4308    388.222    796.068
     0    172.267    178.989    153.137
84.2168    55.7395    4.54389          0
30.1042          0   -13.9315    123.481
129.344    70.8821    54.8537   -8.39166
     0     197.94    97.4388    770.743
217.227    225.907    86.2506    51.2434
     0    116.637    141.063    93.4505
-33.6708          0    655.357    122.723
8.5647          0    167.384    254.902
588.29    -8.5452          0   -70.3225
39.7413   -10.8475          0    231.996
327.745     699.61    -29.602          0
67.8269    37.8128   -30.7696          0
360.526    439.276    628.141   -36.0007
-2.29024    68.7561    52.2916   -21.5845
125.565    47.7142          0    193.054
133.662    116.575    26.8274          0
139.707    165.405    96.8852    26.7167
221.637    258.738    84.8256    54.2643
-18.2649   -34.8866    585.532    318.947
     0    2.48686     211.74    76.6603
     0    766.921    26.5061    623.344
206.352    160.221    150.096    114.838
-37.7703          0    200.931    1.49491
-3.08568          0    6.13958    223.889
62.4522    2.99485          0    173.212
75.8723    44.1407    27.5277          0
     0    141.066    155.178    145.194
298.614   -98.7614          0    323.483
600.229    478.486    23.4052          0
32.5113          0    207.983    188.308
124.375   -24.6943          0    237.091
-2.66319    71.1156    53.6128    12.4483
257.622      6.744    56.5033    60.5568
     0    229.399    145.351    50.3401
271.584    304.166    755.247   -27.0927
118.381    122.216   -19.4142          0
-13.7906          0    216.054    229.522
53.5145    5.54774          0   -5.82003
648.171    82.0209    644.824     397.56
180.424    141.382    148.397   -48.1696
80.448    42.2636   -1.15021          0
     0    65.0958    422.959    684.447
222.155    175.785    135.043    126.561
223.901    67.8264    43.6662     4.1231
236.395    137.669    77.1364    40.6829
-29.2652          0    560.127    85.1803
     0    115.897    145.375    60.3489
-17.5332          0    237.052   -10.5542
499.798    9.37554          0     299.16
56.3777   -24.4827          0    139.165
63.8254    42.6994    5.70532          0
     0    -28.701   -4.06227    154.624
337.605     660.68    -28.063          0
93.422          0    187.838    155.877
164.83    17.2606          0    102.328
131.511    107.173    27.3063          0
151.268    64.3761    50.2538    19.2928
5.60736   -3.04675    68.1928     53.503
77.6406    609.464    399.754   -68.6249
141.245    147.345    125.291     -42.33
     0    217.258    179.827    73.5991
340.08    238.911    552.878    528.094
     0    89.6108     74.833    642.637
5.87447          0    181.594    139.403
38.144   -9.06293          0    164.316
73.8333    41.2388   -12.5681          0
204.158    32.8448      61.08  -0.582094
28.7765          0    1090.15    743.873
91.816    126.115    136.192    116.987
491.416    26.7796          0    267.479
129.952    145.857    59.5746    51.6271
     0    230.506    127.627    138.018
396.879     825.14     25.449          0
     0    220.689    222.144    81.0732
26.0409          0   -29.6759    185.568
47.2985    407.794    584.649    36.5633
-18.2708          0    219.044    217.356
97.6413   -67.9577          0   -9.08261
153.891    107.147    21.4022          0
724.846   -8.23449    491.898    482.909
     0    654.581    7.55405    360.084
55.0669    27.5037          0   -7.61529
87.5999    45.7731    23.9386          0
1.61099    71.1326    53.6832   -15.9896
2.27681          0    1197.97    763.426
33.1588     236.83    72.7748     40.401
     0    91.4131    168.571    62.1417
-19.5451          0    264.596   -2.22778
45.3529   -26.2554          0     263.27
54.3217    56.3644   -41.4864          0
206.387     171.32    84.0227    28.1482
453.034    20.4072          0    41.0942
218.572    42.2471    60.9255   0.736965
-12.649    252.083    123.929    114.841
170.404    137.497    67.0989    40.3043
     0    49.4236    208.537    107.181
16.5775          0    169.686    143.828
652.987    366.639    29.0785          0
41.3366    641.369     373.14    25.7101
     0    139.363    165.522    154.031
67.1085    4.28586    724.016    286.237
-67.3677          0   -2.96634    186.275
89.8631    56.5994          0     36.308
41.4304    29.9163          0    219.317
60.4402    51.6209   -8.05833          0
72.6161    115.424  -0.467638          0
170.649    130.113    110.862   -28.9894
202.974    170.821    111.144    173.394
135.845     59.823     45.225    -21.193
     0   -10.6493    128.124    178.197
204.173    204.639    85.4955    59.6282
     0    166.772    145.438    64.0885
4.95679          0    11.8164    170.102
     0    174.822    399.579    349.644
216.131    25.2534          0   -14.0293
10.0079          0    286.862    198.818
134.57    118.402   -25.8339          0
50.0486   -27.1129          0    189.101
71.2693    51.7942    23.1722          0
187.27    190.644     87.542   -31.6624
162.183    178.066      180.3     90.106
-3.78285    69.5133    52.6819   -18.6811
     0    71.7961    140.305    140.349
-0.480663          0    510.305    108.846
24.4967          0    3.23591    125.444
135.695   -47.1104          0    185.944
470.272    36.8191          0    692.484
79.5459    150.025    87.9097    40.8606
     0   -4.14026    226.042    66.8353
-9.57667          0     269.66    1.32361
55.9444    49.8638          0    177.754
615.837    385.008    19.2578          0
131.539    146.354   -12.2667          0
63.9001    40.1769    -29.725          0
152.854    126.197    138.901   -17.3727
7.00253    227.702    148.069    130.472
189.39    72.6588    58.8961  -0.590718
92.9688    351.301    750.455    28.3629
236.663    128.855    58.7691    42.3523
     0    64.0349     147.16    122.276
-18.6285          0   -11.9515    213.513
94.3151   -27.4796          0    228.248
     0    149.222     139.77     65.109
160.699    88.3438    59.1568          0
6.2178          0    252.544   -3.03933
290.356    224.132    703.355    365.476
148.559    148.662    87.9526   -71.7815
-19.874    194.822    178.863    91.6126
45.8409    26.3251          0    171.046
72.0335    42.1834   0.449649          0
234.509    72.7974    41.5825    4.64336
181.577    145.049     71.756    45.6585
     0   -5.11962    153.392    92.3487
     0    41.3972    207.579    80.5228
-25.7655          0     162.79    236.515
41.9034   -13.3804          0    -9.8899
61.9359    49.6144    7.63537          0
     0    141.677    205.209    407.775
140.448    68.2393    56.2317   -2.87701
236.88    142.446    71.5404    36.9078
13.0965          0    15.3375     146.73
     0    142.531    146.786    68.7194
5.85455          0  -0.110609    222.706
105.114    22.4717          0    190.754
45.0938   -22.4622          0    158.547
27.0491          0    568.687    122.725
400.319   -25.6579          0    204.831
409.662    389.189  -0.323644          0
83.9531    38.6419    -28.621          0
210.09    799.314    307.639   -92.0668
151.106    76.3539    55.8648   -11.2535
152.103    95.6179   -69.8154          0
189.881    168.592    70.8132    44.9752
77.1867    358.932    689.111    367.147
     0    208.862     176.24    64.8214
4.82167          0    204.734    253.176
48.1567    -4.2331          0    4.26904
74.1496     63.925    3.26151          0
     0    166.834    227.442    629.296
136.816     135.56    115.551    24.6008
115.759    153.371    175.038     93.794
130.505    64.0974    45.0222  -0.618799
187.099    195.296    77.8269     43.173
60.7053          0    339.577    119.711
     0   -9.28623    188.984    67.6697
12.3096          0    3.28496    234.695
     0    -8.7913    211.618    169.252
50.6093    31.5212          0    74.5741
5.49367          0    96.1014    155.702
59.8883    54.8697    17.0656          0
90.0935   -29.8435          0    172.586
164.084    67.5133    53.5425    28.5612
88.4888    224.349    26.5431          0
41.385    232.687    80.3714    38.6605
144.877    92.8128    195.846    2.22869
175.809     159.38    166.932    93.6756
     0   -2.66155    189.715    70.1294
-23.9329          0    82.2841    159.404
     0    203.028     142.33    91.4313
6.57373          0   -35.6893    201.533
580.763   -20.2706          0    260.232
96.5024   -67.7647          0    115.589
49.6533    4.64459          0    184.943
66.028    44.3583    1.48551          0
106.585    133.694    51.7336          0
163.302    195.883    94.8303    5.11134
147.064    70.7438    56.2558   -1.48029
687.03    509.267   -19.3377          0
230.386    251.583    76.9151    49.4109
     0    60.4689    213.624    92.1221
155.147    137.473    147.808    102.285
243.025    668.679    339.802      61.17
     0   -12.5941    162.718    182.064
-19.7356          0    79.6946    185.901
47.6636    -29.542          0     46.338
69.6199     40.192    -11.574          0
131.794    86.8363    52.6252   -11.3617
-93.2659          0    168.216    162.838
90.2763    59.7096          0    98.0886
155.278    115.907    29.5137          0
196.669    120.578    138.601   -21.2211
239.382     162.87    203.093    89.0573
     0    217.528    148.262    189.205
1.94467          0    164.213    180.574
128.38   -46.9688          0    219.476
171.222    97.4837    35.0269          0
180.853    169.883    417.443    558.582
143.68    168.684    86.8999    27.4053
130.925    147.463    159.478     75.259
     0      9.957    171.421    315.029
     0    243.586    159.917    88.5721
-18.2499          0    150.685    329.664
25.0074          0    174.841    165.226
404.008   -5.08557          0    95.6328
96.7575   -24.9632          0    206.229
168.801    90.1097    56.8618          0
167.832    171.228    88.7294    59.4581
-1.77757    226.605    150.083    128.158
     0    158.066    169.468    155.363
191.266    372.712   0.131504          0
-67.0476          0   -15.3446    234.407
89.1373    32.2076          0   -11.6607
112.369     703.79    349.389   -24.9751
125.218    149.948    14.0144          0
180.917    151.451    111.019    23.7848
114.919    153.976    153.562    97.7885
     0    188.024    169.646    82.8269
-31.5687          0   -19.0163    236.027
95.4877   -67.7737          0    246.729
28.6398    17.4159          0   -6.26721
237.664   -27.4101    310.384    752.068
159.713    131.304   -26.7936          0
36.8459    29.4731    20.1808          0
166.54    178.495    86.6794    3.76299
64.0152    171.057    126.936    111.576
     0    187.565    192.094    152.463
26.7635          0    125.491    170.777
87.496    63.8674          0    110.314
169.71    89.2564    64.7126          0
 327.5    152.525    110.611    21.0288
180.53    145.016    136.768    108.763
     0    187.203    139.386    90.7944
7.46496          0    213.081    179.364
98.2529   0.828072          0    45.1297
124.083    111.379   -19.6885          0
159.423    134.892    127.877   -9.25682
225.188    168.845    92.7088    221.811
     0   -2.83738    244.676    194.077
     0    233.836    324.098    468.135
10.426          0    110.326   -55.3548
-31.0689          0   -29.8628    108.418
77.8953    86.7444          0    91.7521
151.741    115.001   -27.8802          0
181.859     196.76    97.9117   -29.4974
129.458    163.988    186.511    91.8127
     0    129.711    184.729    183.235
-32.4105          0   -11.5559    229.056
669.793    7.78341          0    325.806
90.0607    4.20093          0    97.8333
185.409    86.1221   -27.0927          0
351.381    931.783     21.768          0
191.726    129.436    118.875   -24.6683
81.0427    153.466    108.707    153.051
     0    99.2251    150.122    161.474
-70.6083          0    63.4194    121.769
78.3949    93.6313          0    215.974
115.074    138.168    44.9516          0
161.951     127.24    149.421    -17.419
66.5737    166.528    191.397    79.0336
220.677    426.234    630.002   -50.0112
     0    108.408    143.061    163.032
93.4821    247.583    345.982    764.263
  4. please simplify the steps as much as possible so they do not require additional resources to run, such as a private dataset.

Expected behavior:

export to onnx model

Environment:

torch,torchvision,detectron2 are all installed from binary.

Provide your environment information using the following command:

----------------------  --------------------------------------------------------------------------------------
sys.platform            linux
Python                  3.6.10 |Anaconda, Inc.| (default, Jan  7 2020, 21:14:29) [GCC 7.3.0]
numpy                   1.18.1
detectron2              0.2 @/home/jushi/d2export/detectron2/detectron2
detectron2._C           failed to import
Compiler                c++ (Ubuntu 5.4.0-6ubuntu1~16.04.12) 5.4.0 20160609
CUDA compiler           Cuda compilation tools, release 10.1, V10.1.168
DETECTRON2_ENV_MODULE   <not set>
PyTorch                 1.6.0+cu101 @/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torch
PyTorch debug build     False
GPU available           True
GPU 0                   GeForce RTX 2080 Ti
CUDA_HOME               /usr/local/cuda-10.1
Pillow                  7.2.0
torchvision             0.7.0+cu101 @/home/jushi/.conda/envs/detectron/lib/python3.6/site-packages/torchvision
torchvision arch flags  sm_35, sm_50, sm_60, sm_70, sm_75
fvcore                  0.1.1.post20200716
cv2                     4.2.0
----------------------  --------------------------------------------------------------------------------------
PyTorch built with:
  - GCC 7.3
  - C++ Version: 201402
  - Intel(R) Math Kernel Library Version 2019.0.5 Product Build 20190808 for Intel(R) 64 architecture applications
  - Intel(R) MKL-DNN v1.5.0 (Git Hash e2ac1fac44c5078ca927cb9b90e1b3066a0b2ed0)
  - OpenMP 201511 (a.k.a. OpenMP 4.5)
  - NNPACK is enabled
  - CPU capability usage: AVX2
  - CUDA Runtime 10.1
  - NVCC architecture flags: -gencode;arch=compute_37,code=sm_37;-gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_75,code=sm_75
  - CuDNN 7.6.3
  - Magma 2.5.2
  - Build settings: BLAS=MKL, BUILD_TYPE=Release, CXX_FLAGS= -Wno-deprecated -fvisibility-inlines-hidden -DUSE_PTHREADPOOL -fopenmp -DNDEBUG -DUSE_FBGEMM -DUSE_QNNPACK -DUSE_PYTORCH_QNNPACK -DUSE_XNNPACK -DUSE_VULKAN_WRAPPER -O2 -fPIC -Wno-narrowing -Wall -Wextra -Werror=return-type -Wno-missing-field-initializers -Wno-type-limits -Wno-array-bounds -Wno-unknown-pragmas -Wno-sign-compare -Wno-unused-parameter -Wno-unused-variable -Wno-unused-function -Wno-unused-result -Wno-unused-local-typedefs -Wno-strict-overflow -Wno-strict-aliasing -Wno-error=deprecated-declarations -Wno-stringop-overflow -Wno-error=pedantic -Wno-error=redundant-decls -Wno-error=old-style-cast -fdiagnostics-color=always -faligned-new -Wno-unused-but-set-variable -Wno-maybe-uninitialized -fno-math-errno -fno-trapping-math -Werror=format -Wno-stringop-overflow, PERF_WITH_AVX=1, PERF_WITH_AVX2=1, PERF_WITH_AVX512=1, USE_CUDA=ON, USE_EXCEPTION_PTR=1, USE_GFLAGS=OFF, USE_GLOG=OFF, USE_MKL=ON, USE_MKLDNN=ON, USE_MPI=OFF, USE_NCCL=ON, USE_NNPACK=ON, USE_OPENMP=ON, USE_STATIC_DISPATCH=OFF,

Corresponding files:

There are model file, config file and test image. https://drive.google.com/file/d/1TEk55eHwb41Fn43ypJ5stLDNkDxwr2AJ/view?usp=sharing

ppwwyyxx commented 4 years ago

It turns out that we have only added support of Rotated R-CNN for non-FPN models due to missing operators in caffe2.