Update README.md
Browse files
README.md
CHANGED
|
@@ -45,7 +45,7 @@ The data curation pipeline leading to the clean videos in the Surg-3M dataset is
|
|
| 45 |
|
| 46 |
Usage
|
| 47 |
--------
|
| 48 |
-
Video classification models are employed in the step 2 of the data curation pipeline to classify a video storyboard as either surgical or non-surgical, the models usage is as follows:
|
| 49 |
```python
|
| 50 |
import torch
|
| 51 |
from PIL import Image
|
|
@@ -72,7 +72,7 @@ Video classification models are employed in the step 2 of the data curation pipe
|
|
| 72 |
outputs = net(img_tensor)
|
| 73 |
```
|
| 74 |
|
| 75 |
-
Frame classification models are used in the step 3 of the data curation pipeline to classify a frame as either surgical or non-surgical, the models usage is as follows:
|
| 76 |
|
| 77 |
```python
|
| 78 |
import torch
|
|
@@ -99,7 +99,7 @@ Frame classification models are used in the step 3 of the data curation pipeline
|
|
| 99 |
outputs = net(img_tensor)
|
| 100 |
```
|
| 101 |
|
| 102 |
-
Non-surgical object detection models are used to obliterate the non-surgical region in the surgical frames (e.g. user interface information), the models usage is as follows:
|
| 103 |
|
| 104 |
```python
|
| 105 |
import torch
|
|
|
|
| 45 |
|
| 46 |
Usage
|
| 47 |
--------
|
| 48 |
+
**Video classification models** are employed in the step 2 of the data curation pipeline to classify a video storyboard as either surgical or non-surgical, the models usage is as follows:
|
| 49 |
```python
|
| 50 |
import torch
|
| 51 |
from PIL import Image
|
|
|
|
| 72 |
outputs = net(img_tensor)
|
| 73 |
```
|
| 74 |
|
| 75 |
+
**Frame classification models** are used in the step 3 of the data curation pipeline to classify a frame as either surgical or non-surgical, the models usage is as follows:
|
| 76 |
|
| 77 |
```python
|
| 78 |
import torch
|
|
|
|
| 99 |
outputs = net(img_tensor)
|
| 100 |
```
|
| 101 |
|
| 102 |
+
**Non-surgical object detection models** are used to obliterate the non-surgical region in the surgical frames (e.g. user interface information), the models usage is as follows:
|
| 103 |
|
| 104 |
```python
|
| 105 |
import torch
|