Commit d5f9c39a authored by Meet Narendra's avatar Meet Narendra 💬

Removed fmaps

parent 1af71c2e
...@@ -52,21 +52,6 @@ class FeatureMaps(): ...@@ -52,21 +52,6 @@ class FeatureMaps():
layer_num+=1 layer_num+=1
return fmaps return fmaps
def get_fmaps_style(self,img,layer=[0,5,10,19,28]):
'''
Function which will pass the image through the model and get the respective fmaps
@params
img: numpy image f64
layer: list
'''
fmaps = []
layer_num = 0
for layer_i in self.model.features:
img = layer_i(img)
if layer_num in layer:
fmaps.append(img)
layer_num+=1
return fmaps
''' '''
if __name__ == "__main__": if __name__ == "__main__":
fmap = FeatureMaps() fmap = FeatureMaps()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment