OrgLion-ML
/
deep-text-recognition-benchmark
mirror of https://github.com/clovaai/deep-text-recognition-benchmark.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Update transformation.py

This commit is contained in:
Baek JeongHun 2020-01-17 00:12:18 +09:00 committed by GitHub
parent 382a9b50f3
commit 78d7ed96a9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/transformation.py
View File

@ -4,6 +4,7 @@ import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
class TPS_SpatialTransformerNetwork(nn.Module): class TPS_SpatialTransformerNetwork(nn.Module):
""" Rectification Network of RARE, namely TPS based STN """ """ Rectification Network of RARE, namely TPS based STN """