Delete SimSwap/models/fs_model.py

SimSwap/models/fs_model.py

@@ -1,36 +0,0 @@
-
-import torch
-from .base_model import BaseModel
-from . import networks
-
-class FSModel(BaseModel):
-    def name(self):
-        return 'FSModel'
-
-    def initialize(self, opt):
-        BaseModel.initialize(self, opt)
-        self.opt = opt
-        self.device = torch.device("cpu")
-
-        self.netG = networks.define_G(opt, self.device)
-        self.netArc = networks.define_arcface(opt, self.device)
-
-        self.load_network(self.netG, 'G', opt.which_epoch, opt.load_pretrain)
-        self.load_network(self.netArc, 'arc', opt.which_epoch, opt.load_pretrain)
-
-        self.netG.eval()
-        self.netArc.eval()
-
-    def set_input(self, opt, id_feature):
-        self.image_paths = opt.pic_a_path
-        self.id_feature = id_feature
-
-    def forward(self):
-        with torch.no_grad():
-            self.fake_B = self.netG(self.id_feature, self.image_paths)
-
-    def test(self):
-        self.forward()
-
-    def get_current_visuals(self):
-        return {'synthesized_image': self.fake_B}