Browse Source
        
      
      changed the operator
      
        Signed-off-by: krishna katyal <krishna.katyal@zilliz.com>
      
      
        main
      
      
     
    
      
        
          
            
            krishna katyal
          
          4 years ago
          
         
        
        
       
      
     
    
    
	
		
			
				 1 changed files with 
1 additions and 
1 deletions
			 
			
		 
		
			
				- 
					
					
					 
					deepface.py
				
 
			
		
		
			
			
			
			
			
			
				
				
					
						
							
								
									
	
		
			
				
					| 
						
						
							
								
							
						
						
					 | 
				
				 | 
				
					@ -19,7 +19,7 @@ class DeepfaceFaceEmbedding(NNOperator): | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					      self.model_name=model_name | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					  @arg(1, to_image_color('RGB')) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					  def __call__(self,img: Image) -> np.ndarray: | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					      open_cv_image = np.array(Image)  | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					      open_cv_image = np.array(img)  | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					      # Convert RGB to BGR  | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					      open_cv_image = open_cv_image[:, :, ::-1].copy()  | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					      # img.to_ndarray() | 
				
			
			
		
	
	
		
			
				
					| 
						
							
								
							
						
						
						
					 | 
				
				 | 
				
					
  |