Commit ed8bb266 authored by dzy7e's avatar dzy7e
Browse files

infer batch

parent ee48ae64

zoo/ccip/demo.py

+1 −0
Original line number Diff line number Diff line
@@ -44,6 +44,7 @@ class Infer: 
            device=self.device,

            fp16=args.fp16

        )

        self.model.eval()



    def load_img(self, path):

        image = load_image(path, mode='RGB')