Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/WarBean/tps_stn_pytorch
/ functions
Functions
19 in github.com/WarBean/tps_stn_pytorch
⨍
Functions
19
◇
Types & classes
6
↓ 4 callers
Method
__init__
(self, num_output)
mnist_model.py:14
↓ 3 callers
Function
grid_sample
(input, grid, canvas = None)
grid_sample.py:6
↓ 2 callers
Function
compute_partial_repr
(input_points, control_points)
tps_grid_gen.py:9
↓ 1 callers
Function
test
(epoch)
mnist_train.py:69
↓ 1 callers
Function
train
(epoch)
mnist_train.py:48
Method
__init__
(self)
mnist_model.py:33
Method
__init__
(self, grid_height, grid_width, target_control_points)
mnist_model.py:42
Method
__init__
(self, grid_height, grid_width, target_control_points)
mnist_model.py:58
Method
__init__
(self, args)
mnist_model.py:73
Method
__init__
(self, target_height, target_width, target_control_points)
tps_grid_gen.py:25
Method
forward
(self, x)
mnist_model.py:22
Method
forward
(self, x)
mnist_model.py:37
Method
forward
(self, x)
mnist_model.py:51
Method
forward
(self, x)
mnist_model.py:66
Method
forward
(self, x)
mnist_model.py:97
Method
forward
(self, source_control_points)
tps_grid_gen.py:62
Function
get_model
(args)
mnist_model.py:106
Function
get_test_loader
(args)
data_loader.py:24
Function
get_train_loader
(args)
data_loader.py:7