self.eanet_trainer = EANetTrainer(args=args) feature= self.eanet_trainer.infer_one_im(im=img_rgb, squeeze=False)