Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- final List<ClarifaiOutput<Concept>> predictionResults2 = // Concept zamieniony na prediction
- client.getDefaultModels().generalModel()//.predict("kulak")//getModelByID("kulak") //getDefaultModels().generalModel() // You can also do client.getModelByID("id") to get custom models
- .predict()
- .withInputs(
- ClarifaiInput.forImage(ClarifaiImage.of("http://pre01.deviantart.net/6be4/th/pre/i/2012/001/6/1/girl_with_the_glasses_by_suannystock-d4kx0l4.jpg"))
- )
- .executeSync()
- .get();
- HashMap<String,Double> map_tag = new HashMap<String,Double>();
- for (Concept temp : predictionResults2.get(0).data()) {
- if(temp.value() > 0.92)
- {
- if(temp.name().equals("eyeglasses"))
- map_tag.put("glasses",(double) temp.value());
- if(temp.name().equals("eyewear"))
- map_tag.put("glasses",(double) temp.value());
- else if(temp.name().equals("woman"))
- map_tag.put("female",(double) temp.value());
- else if(temp.name().equals("child"))
- map_tag.put("teen",(double) temp.value());
- else if(temp.name().equals("man"))
- map_tag.put("male",(double) temp.value());
- else if(temp.name().equals("business"))
- map_tag.put("garnitur",(double) temp.value());
- else if(temp.name().equals("tie"))
- map_tag.put("tie",(double) temp.value());
- else if(temp.name().equals("cap"))
- map_tag.put("czapka_daszek",(double) temp.value());
- else if(temp.name().equals("fashion"))
- map_tag.put("fashion",(double) temp.value());
- }
- }
- // najpierw custom model
- for (Prediction temp : predictionResults.get(0).data())
- {
- if(temp.asConcept().value() > 0.60)
- { //System.out.println(temp.asConcept().name());
- if(temp.asConcept().name().equals("female") && !map_tag.containsKey("male") ) // tag to female, sprawdz czy nie dodalismy juz male bo ma wieksze prawdopodobienstwo
- map_tag.put("female",(double) temp.asConcept().value());
- else if(temp.asConcept().name().equals("male") && !map_tag.containsKey("female") ) // tag to male, sprawdz czy nie dodalismy juz male bo ma wieksze prawdopodobienstwo
- map_tag.put("male",(double) temp.asConcept().value());
- else if(temp.asConcept().name().equals("middle") && !map_tag.containsKey("old") && !map_tag.containsKey("teen")) // tag to female, sprawdz czy nie dodalismy juz male bo ma wieksze prawdopodobienstwo
- map_tag.put("middle",(double) temp.asConcept().value());
- else if(temp.asConcept().name().equals("teen") && !map_tag.containsKey("old") && !map_tag.containsKey("middle") ) // tag to female, sprawdz czy nie dodalismy juz male bo ma wieksze prawdopodobienstwo
- map_tag.put("teen",(double) temp.asConcept().value());
- else if(temp.asConcept().name().equals("old") && !map_tag.containsKey("teen") && !map_tag.containsKey("middle") ) // tag to female, sprawdz czy nie dodalismy juz male bo ma wieksze prawdopodobienstwo
- map_tag.put("old",(double) temp.asConcept().value());
- else if(temp.asConcept().name().equals("garnitur"))
- map_tag.put("suit",(double) temp.asConcept().value());
- else // jak nie jest z zadnej grupy np. wiek, plec, i ma > 60.0 pewnosc to dodajemy
- map_tag.put(temp.asConcept().name(),(double) temp.asConcept().value());
- }
- }
- System.out.println("Przekazane tagi: ");
- for (String temp : map_tag.keySet()) {
- System.out.println(temp);
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement