CSAILVision / CSAILVision/semantic-segmentation-pytorch
increase the fps
Open
- Dominant language
- Python
- Stars
- 5.1k
- Forks
- 1.1k
- PR merge metrics
- No merged PRs in 30d
Description
Replace lines 56-56 in "test.py" with the following:
```
# scores = torch.zeros(1, cfg.DATASET.num_class, segSize[0], segSize[1])
# scores = async_copy_to(scores, gpu)
```
to
` scores = torch.zeros(1, cfg.DATASET.num_class, segSize[0], segSize[1], device=torch.device("cuda", gpu))`
This change is to make a zeros tensor on the GPU.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.