I want to train on my own dataset, how should I do it?

WarBean / tps_stn_pytorch

PyTorch implementation of Spatial Transformer Network (STN) with Thin Plate Spline (TPS)

926 stars 155 forks source link

Open liuzhuang1024 opened 5 years ago