towhee
/
            
              image-embedding-resnet50
              
                
                
            
          copied
      
        
          Browse Source
        
      
      
	
		
	
      
        
    Signed-off-by: shiyu22 <shiyu.chen@zilliz.com> Add image_embedding_resnet50.py Signed-off-by: shiyu22 <shiyu.chen@zilliz.com> Add image_embedding_resnet50.py Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>main
				 1 changed files with 21 additions and 0 deletions
			
			
		@ -0,0 +1,21 @@ | 
			
		|||||
 | 
				from towhee import pipe, ops, AutoPipes, AutoConfig | 
			
		||||
 | 
				
 | 
			
		||||
 | 
				@AutoConfig.register | 
			
		||||
 | 
				class ImageEmbeddingConfig: | 
			
		||||
 | 
				    """ | 
			
		||||
 | 
				    For UT | 
			
		||||
 | 
				    """ | 
			
		||||
 | 
				    def __init__(self): | 
			
		||||
 | 
				        self.model = 'resnet50' | 
			
		||||
 | 
				
 | 
			
		||||
 | 
				@AutoPipes.register | 
			
		||||
 | 
				def ImageEmbedding(config): | 
			
		||||
 | 
				    """ | 
			
		||||
 | 
				    For UT | 
			
		||||
 | 
				    """ | 
			
		||||
 | 
				    return ( | 
			
		||||
 | 
				        pipe.input('path') | 
			
		||||
 | 
				        .map('path', 'img', ops.image_decode()) | 
			
		||||
 | 
				        .map('img', 'embedding', ops.image_embedding.timm(model_name=config.model)) | 
			
		||||
 | 
				        .output('embedding') | 
			
		||||
 | 
				    ) | 
			
		||||
					Loading…
					
					
				
		Reference in new issue