画像のカラー抽出を頑張った話。

#=== init
dir_path = "/img/test"


for current_dir, sub_dirs, files_list in os.walk(dir_path): 
  for file_name in files_list:
    path = os.path.join(current_dir,file_name)

    # png,PNG,jpg,JPG,jpeg,JPEG,gif,ico,webp,svg
    flag = path.endswith('.png') or path.endswith('.PNG') or path.endswith('.jpg') or path.endswith('.JPG') or path.endswith('.jpeg') or path.endswith('.JPEG') or path.endswith('.gif') or path.endswith('.ico') or path.endswith('.webp') or path.endswith('.svg')
    
    if flag:
      exact_color(path, 900, 36, 2.5)

Copy

def exact_color(input_image, resize, tolerance, zoom):

    #=== Dataframeの作成
    # svgファイルだった場合、pngに変換してチェックする。svg以外の時はリサイズの判定を挟む。
    if input_image.endswith('.svg'):
      drawing = svg2rlg(input_image)
      renderPM.drawToFile(drawing, "output_svg.png", fmt="PNG")
      img_real = 'output_svg.png'
    else:
      # リサイズを実施
      output_width = resize
      img = Image.open(input_image)
      if img.size[0] >= resize:
        width_per_c = (output_width / float(img.size[0]))
        hsize = int((float(img.size[1]) * float(width_per_c)))
        img = img.resize((output_width, hsize), Image.ANTIALIAS)
        root, ext = os.path.splitext(input_image)
        resize_name = 'resize_img' + ext
        img.save(resize_name)
      else:
        resize_name = input_image
      img_real = resize_name

    # カラーの取得
    colors_x = extcolors.extract_from_path(img_real, tolerance=tolerance, limit=15)

    # 取得したカラーデータの形式を整形
    colors_pre_list = str(colors_x).replace('([(', '').split(', (')[0:-1]
    df_rgb = [i.split('), ')[0] + ')' for i in colors_pre_list]

    #=== RGBからHEXコードへの変換
    df_color_up = [rgb2hex(int(i.split(", ")[0].replace("(", "")),
                           int(i.split(", ")[1]),
                           int(i.split(", ")[2].replace(")", ""))) for i in df_rgb]

    #Dataframe生成 
    df_color = pd.DataFrame(df_color_up, columns=['c_code'])

    out_color = df_color['c_code']
    print("path:" + input_image)
    print(out_color)

    # 文字コードに合致していたら、テキストファイルにパスを出力（svgの場合があるので、引数のファイルパスをそのまま出力）
    for ct in out_color:
      if ct == '#1B3B74' or ct =="#004098":
        f = open('result.txt', 'a')
        f.write(input_image + '\n')
        break

Copy

/img/test/index-img-mail-02--wide.png
/img/test/logo_shinndan.png
/img/test/index_img004.jpg

Copy