зеркало из
https://github.com/ssciwr/AMMICO.git
synced 2025-10-29 21:16:06 +02:00
bec845e774
* fixing DemoNotebook * fixing DemoNotebook2 * rearrange the code and text * continiue changing demo notebook * continiue changing demo notebook2 * fixing bug on a100 in google colab * change order of tests and do not test for age in faces * add screenshots for gcv api * instructions for google cloud vision API * link instructions in readme * add missing image * trim whitespace --------- Co-authored-by: Inga Ulusoy <inga.ulusoy@uni-heidelberg.de>
36 строки
1.2 KiB
Python
36 строки
1.2 KiB
Python
import ammico.faces as fc
|
|
import json
|
|
import pytest
|
|
|
|
|
|
def test_set_keys():
|
|
ed = fc.EmotionDetector({})
|
|
assert ed.subdict["face"] == "No"
|
|
assert ed.subdict["multiple_faces"] == "No"
|
|
assert ed.subdict["wears_mask"] == ["No"]
|
|
assert ed.subdict["emotion"] == [None]
|
|
with pytest.raises(ValueError):
|
|
fc.EmotionDetector({}, emotion_threshold=150)
|
|
with pytest.raises(ValueError):
|
|
fc.EmotionDetector({}, emotion_threshold=-50)
|
|
with pytest.raises(ValueError):
|
|
fc.EmotionDetector({}, race_threshold=150)
|
|
with pytest.raises(ValueError):
|
|
fc.EmotionDetector({}, race_threshold=-50)
|
|
|
|
|
|
def test_analyse_faces(get_path):
|
|
mydict = {
|
|
"filename": get_path + "IMG_2746.png",
|
|
}
|
|
mydict.update(fc.EmotionDetector(mydict).analyse_image())
|
|
|
|
with open(get_path + "example_faces.json", "r") as file:
|
|
out_dict = json.load(file)
|
|
# delete the filename key
|
|
mydict.pop("filename", None)
|
|
# delete the age key, as this is conflicting - gives different results sometimes
|
|
mydict.pop("age", None)
|
|
for key in mydict.keys():
|
|
assert mydict[key] == out_dict[key]
|