7e0249a084341cf2e658b928bc91768b2db036d1,torchreid/engine/engine.py,Engine,visactmap,#Engine#Any#Any#Any#Any#Any#,297

Before Change

            - Zhou et al. Omni-Scale Feature Learning for Person Re-Identification. ICCV, 2019.
        
        self.model.eval()
        if not hasattr(self.model, "featuremaps"):
            raise AttributeError("Model must have method featuremaps(), which returns the feature maps "
                                 "of shape (b, c, h, w)")
        
        imagenet_mean = [0.485, 0.456, 0.406]
        imagenet_std = [0.229, 0.224, 0.225]
        tensor2pil = torchvision.transforms.ToPILImage()

After Change

                
                // forward to get convolutional feature maps
                try:
                    outputs = self.model(imgs, output_featuremaps_only=True)
                except TypeError:
                    raise TypeError("forward() got unexpected keyword argument "output_featuremaps_only". " \
                                    "Please add output_featuremaps_only as an input argument to forward(). When " \
                                    "output_featuremaps_only=True, return feature maps only.")

Instances