logo
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions

87 lines
1.7 KiB

# Filter Tiny Segments
2 years ago
*author: Chen Zhang*
<br />
## Description
This operator can filter tiny detected segments with format of list of `[start_second_1, start_second_2, end_second_1, end_second_2]`
<br />
## Code Example
```python
from towhee.dc2 import pipe, ops, DataCollection
p = (
pipe.input('pred') \
.map('pred', 'filtered_pred', ops.video_copy_detection.filter_tiny_segments(filter_s_thresh=20)) \
.output('pred', 'filtered_pred')
)
DataCollection(p([[0, 0, 100, 100], [0, 0, 10, 10], [0, 0, 60, 10]])).show()
```
![](result.png)
```python
from towhee.dc2 import pipe, ops, DataCollection
p = (
pipe.input('pred', 'sim_hw') \
.map(('pred', 'sim_hw'), 'filtered_pred', ops.video_copy_detection.filter_tiny_segments(filter_s_thresh=20)) \
.output('pred', 'sim_hw', 'filtered_pred')
)
DataCollection(p([[0, 0, 10, 10]], [11, 11])).show()
```
![](result2.png)
## Factory Constructor
Create the operator via the following factory method
***filter_tiny_segments(filter_s_thresh, segment_len_rate)***
**Parameters:**
***filter_s_thresh:*** *float*
​ Use a thresh to filter detected box which is smaller than it.
***segment_len_rate:*** *float*
​ Filter expect longer then segment_len_rate * video length. Only useful for filter expect near video length segments.
<br />
## Interface
**Parameters:**
***pred_value:*** *List*
​ List of predicted segment second infos of a video pair
***sim_hw:*** *Tuple*
​ Similarity matrix height and weight of a video pair. If sample rate is 1s, sim_hw is also the lengths of these videos.
**Returns:**
***res_pred_list:*** *List*
​ List of filtered predicted segment second infos