emkarcinos / AITech-flats

Projekt magisterski AITech - klasyfikacja stylu wnętrz mieszkań i transfer stylu
Apache License 2.0
2 stars 0 forks source link

Model ma zwracać listę prawdopodobieństw zamiast jednej odpowiedzi #46

Open emkarcinos opened 1 year ago

emkarcinos commented 1 year ago

Zamiast wyliczać metryki tylko dla przewidzianej klasy z największym prawdopodobieństwem, chcemy wybrać x najwyższych prawdopodobieństw i liczyć je w ten sposób. Na razie nie będzie miało to większego sensu, bo dataset na którym pracujemy ma tylko jeden label dla każdego obrazka, natomiast w przyszłości chcemy pracować z wieloma labelami per zdjęcie.

Trzeba zaktualizować kod liczący metryki i inferencję.