mirror of
https://github.com/hzwer/ECCV2022-RIFE.git
synced 2025-12-16 08:27:45 +01:00
Move HD model
This commit is contained in:
@@ -85,17 +85,17 @@ if torch.cuda.is_available():
|
||||
|
||||
try:
|
||||
try:
|
||||
from model.RIFE_HDv2 import Model
|
||||
from model.oldmodel.RIFE_HDv2 import Model
|
||||
model = Model()
|
||||
model.load_model(args.modelDir, -1)
|
||||
print("Loaded v2.x HD model.")
|
||||
except:
|
||||
from train_log.RIFE_HDv3 import Model
|
||||
from train_log.oldmodel.RIFE_HDv3 import Model
|
||||
model = Model()
|
||||
model.load_model(args.modelDir, -1)
|
||||
print("Loaded v3.x HD model.")
|
||||
except:
|
||||
from model.RIFE_HD import Model
|
||||
from model.oldmodel.RIFE_HD import Model
|
||||
model = Model()
|
||||
model.load_model(args.modelDir, -1)
|
||||
print("Loaded v1.x HD model")
|
||||
|
||||
@@ -6,7 +6,7 @@ import torch.optim as optim
|
||||
import itertools
|
||||
from model.warplayer import warp
|
||||
from torch.nn.parallel import DistributedDataParallel as DDP
|
||||
from model.IFNet_HD import *
|
||||
from model.oldmodel.IFNet_HD import *
|
||||
import torch.nn.functional as F
|
||||
from model.loss import *
|
||||
|
||||
@@ -6,7 +6,7 @@ import torch.optim as optim
|
||||
import itertools
|
||||
from model.warplayer import warp
|
||||
from torch.nn.parallel import DistributedDataParallel as DDP
|
||||
from model.IFNet_HDv2 import *
|
||||
from model.oldmodel.IFNet_HDv2 import *
|
||||
import torch.nn.functional as F
|
||||
from model.loss import *
|
||||
|
||||
Reference in New Issue
Block a user