Skip to content

1.0.5

Compare
Choose a tag to compare
@PINTO0309 PINTO0309 released this 01 Sep 01:11
· 17 commits to main since this release
  • Support for SequenceConstruct
    INFO: file: yolopv2_192x320_test.onnx
    INFO: providers: ['CUDAExecutionProvider', 'CPUExecutionProvider']
    INFO: input_name.1: input shape: [1, 3, 192, 320] dtype: float32
    INFO: test_loop_count: 10
    INFO: total elapsed time:  76.85065269470215 ms
    INFO: avg elapsed time per pred:  7.685065269470215 ms
    INFO: output_name.1-0: pred-0 shape: [1, 255, 24, 40] dtype: float32
    INFO: output_name.1-1: pred-1 shape: [1, 255, 12, 20] dtype: float32
    INFO: output_name.1-2: pred-2 shape: [1, 255, 6, 10] dtype: float32
    INFO: output_name.2: anchor_grid0 shape: [1, 3, 1, 1, 2] dtype: float32
    INFO: output_name.3: anchor_grid1 shape: [1, 3, 1, 1, 2] dtype: float32
    INFO: output_name.4: anchor_grid2 shape: [1, 3, 1, 1, 2] dtype: float32
    INFO: output_name.5: seg shape: [1, 2, 192, 320] dtype: float32
    INFO: output_name.6: ll shape: [1, 1, 192, 320] dtype: float32
    
    image
  • Add short form
    usage:
      sit4onnx [-h]
      -if INPUT_ONNX_FILE_PATH
      [-b BATCH_SIZE]
      [-fs DIM0 [DIM1 DIM2 ...]]
      [-tlc TEST_LOOP_COUNT]
      [-oep {tensorrt,cuda,openvino_cpu,openvino_gpu,cpu}]
      [-ifp INPUT_NUMPY_FILE_PATHS_FOR_TESTING]
      [-ofp]
      [-n]