76 lines
1.6 KiB
Markdown
76 lines
1.6 KiB
Markdown
# sail 基础函数
|
||
|
||
## get_available_tpu_num
|
||
|
||
获取当前设备中可用的TPU数量
|
||
|
||
**接口形式：**
|
||
```python
|
||
def get_available_tpu_num() -> int
|
||
```
|
||
|
||
**返回值说明**:
|
||
|
||
返回当前设备中可用 TPU的数量
|
||
|
||
## set_print_flag
|
||
|
||
设置是否打印程序的计算耗时信息
|
||
|
||
**接口形式:**
|
||
|
||
```python
|
||
def set_print_flag(print_flag: bool) -> None
|
||
```
|
||
|
||
**参数说明**
|
||
|
||
- print_flag:bool
|
||
|
||
print_flag 为 True 时，打印程序的计算主要的耗时信息，否则不打印
|
||
|
||
## set_dump_io_flag
|
||
|
||
设置是否存储输入时间和输出时间。
|
||
|
||
**接口形式:**
|
||
|
||
```python
|
||
def set_dump_io_flag(dump_io_flag: bool) -> None
|
||
```
|
||
|
||
**参数说明**
|
||
|
||
- dump_io_flag: bool
|
||
|
||
dump_io_flag为True时，存储输入数据和输出数据，否则不存储。
|
||
|
||
## set_decoder_env
|
||
|
||
设置Decoder的环境变量，必须在Decoder构造前设置，否则使用默认值。
|
||
|
||
**接口形式:**
|
||
|
||
```python
|
||
def set_decoder_env(env_name: str, env_value: str) -> None
|
||
```
|
||
|
||
**参数说明:**
|
||
|
||
- env_name: str
|
||
|
||
选择设置Decoder的属性名称，可选的属性名称有:
|
||
|
||
- ‘refcounted_frames’ 设置为1时，解码出来的图像需要程序手动释放，为0时由Decoder自动释放。
|
||
- ‘extra_frame_buffer_num’ 设置Decoder的最大缓存帧数
|
||
- ‘rtsp_transport’ 设置RTSP采用的传输协议
|
||
- ‘stimeout’ 设置阻塞超时时间
|
||
- ‘rtsp_flags’ 设置RTSP是否自定义IO
|
||
- ‘buffer_size’ 设置缓存大小
|
||
- ‘max_delay’ 设置最大时延
|
||
- ‘probesize’ 解析文件时读取的最大字节数
|
||
- ‘analyzeduration’ 解析文件时读取的最大时长
|
||
|
||
- env_value: str
|
||
|