Prevod obrazku do csv suboru v pythone – Python – Fórum – Programujte.com

Zdravím.Mám kód ktorý z priečinku obrázkou ľudi vypíše emocie . V console ich vypise napr. angry 0.004 , happy 1.2 atd.Všetky fotky, ktoré mám , treba poslať na analýzu . Výsledky potom treba uložiť do nejakého súboru (najlepšie csv) v ktorom jeden stĺpec bude emócia, ktorá bola rozpoznaná ľuďmi (t.j. názov súboru) a do druhého emócia rozpoznaná Face API. napr. v prípade fotky v priečinku j, joy4 to bude vyzerať nasledovne: j/joy4 joy happiness (keďže systém rozpoznal happiness, viď príloha) súbor bude mať toľko riadkov koľko je fotiek. viď. príloha

Pýtal som sa niekoľko ľudí a nikto my s tým nevedel pomôcť.

Připojen obrázek.

import requests
import os
import time
import pprint
#import csv


BASE_URL = 'https://westcentralus.api.cognitive.microsoft.com/face/v1.0/detect'
headers = {
    'Ocp-Apim-Subscription-Key': 'cccf20b44cb9445a8a4239ce324acfed',
    'Content-Type': 'application/octet-stream'
}
parameters = {
    'returnFaceId': 'true',
    'returnFaceLandmarks': 'false',
    'returnFaceAttributes': 'age,gender,emotion'
}

def post_image(img_data):
    response = requests.post(BASE_URL, params=parameters,
                             headers=headers, data=img_data)
    try:
        return response.json()
    except:
        return None



img_path = '' #nastavit cestu ku priecinku s fotkami (napr. a, b, c)

files = os.listdir(img_path)

for file in files:
    file = os.path.join(img_path,file)
    print(file)
    img_data = open(file, 'rb').read()
    print(post_image(img_data))
    # pprint.pformat(post_image(img_data))
    time.sleep(3)



#with open(file, 'w') as csv_file:
     #   writer = csv.writer(csv_file)
      #  writer.writerow(('age', 'emotion'))
        # pprint.pformat(post_image(img_data))
    
    #return os.path.abspath(img_path)

Tak nejdřív otevřeš writer a pak do něj v tom zpracovávacím cyklu můžeš zapisovat výsledky. Může to vypadat třeba nějak takhle:

with open(file, 'w') as csv_file:
    writer = csv.writer(csv_file)
    writer.writerow(('age', 'emotion'))

    for file in files:
        ...
        response = post_image(img_data))
        writer.writerow((response['age'], response['emotion']))

Když to čtu pozorněji, tak to asi není přesně to, co chceš, ale princip je snad jasný.

#3 gna
Asi ano, hlavne mam problem s tym ako to zakomponovat do mojho kodu aby to fungovalo. Ono to vlastne prechadza priečinok s obrazkami kde môj kód vypíše emócie z fotiek za sebou a ja potrebujem aby tie parametre ukladalo priamo do v csv súbore v takom tvare ako je vyššie napísané.

#4 Lubo
 podle mě je to jen správném provázání těch 2 kódů

GNA Ti to v podstatě vyřešil, otevření CSV a První řádka s nadpisy před cyklus na procházení fotek, a v cyklu fotek pak správně dekódovat ten výsledek v JSON co vrací Post_Image a naformátovat pro řádek v csv

Na začátek řádků s obrázky bych možná přidal jméno souboru