2018-02-23 17:04:28 -05:00
|
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
|
|
|
|
|
|
import java.awt.AWTException;
|
2018-02-24 10:49:49 -05:00
|
|
|
import java.awt.image.BufferedImage;
|
|
|
|
import java.io.File;
|
|
|
|
import java.io.IOException;
|
|
|
|
import java.util.ArrayList;
|
|
|
|
|
|
|
|
import javax.imageio.ImageIO;
|
2018-02-23 17:04:28 -05:00
|
|
|
|
|
|
|
import org.junit.jupiter.api.Test;
|
2018-02-24 10:49:49 -05:00
|
|
|
import org.opencv.core.Rect2d;
|
2018-02-23 17:04:28 -05:00
|
|
|
|
|
|
|
class ObjectDetectorTest {
|
|
|
|
|
|
|
|
@Test
|
2018-02-24 10:49:49 -05:00
|
|
|
void testObjectDetection() throws Exception {
|
2018-02-23 17:04:28 -05:00
|
|
|
ObjectDetector objectDetector = new ObjectDetector();
|
2018-02-24 10:49:49 -05:00
|
|
|
BufferedImage testImage1 = ImageIO.read(new File("/home/dpapp/tensorflow-1.5.0/models/raccoon_dataset/test_images/ironOre_test_9.jpg"));
|
2018-02-24 14:42:05 -05:00
|
|
|
ArrayList<DetectedObject> detectedObjects1 = objectDetector.getObjectsInImage(testImage1, 0.8);
|
2018-02-24 10:49:49 -05:00
|
|
|
ArrayList<DetectedObject> detectedIronOres1 = objectDetector.getObjectsOfClassInList(detectedObjects1, "ironOre");
|
|
|
|
ArrayList<DetectedObject> detectedOres1 = objectDetector.getObjectsOfClassInList(detectedObjects1, "ore");
|
|
|
|
|
|
|
|
assertEquals(3, detectedIronOres1.size());
|
|
|
|
assertEquals(2, detectedOres1.size());
|
|
|
|
//assertDetectedObjectsAreEqual();
|
|
|
|
}
|
|
|
|
|
2018-02-24 14:42:05 -05:00
|
|
|
|
|
|
|
/*@Test
|
|
|
|
void testObjectDetectionInSubImage() throws Exception {
|
|
|
|
ObjectDetector objectDetector = new ObjectDetector();
|
|
|
|
BufferedImage testImage = ImageIO.read(new File("/home/dpapp/tensorflow-1.5.0/models/raccoon_dataset/test_images/ironOre_test_9.jpg"));
|
|
|
|
}*/
|
|
|
|
|
2018-02-24 10:49:49 -05:00
|
|
|
void assertDetectedObjectsAreEqual(DetectedObject obj1, DetectedObject obj2) {
|
2018-02-23 17:04:28 -05:00
|
|
|
|
2018-02-24 10:49:49 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
void assertBoundingBoxesAreEqual(Rect2d bb1, Rect2d bb2) {
|
|
|
|
assertEquals(bb1.x, bb2.x, 3);
|
|
|
|
assertEquals(bb1.y, bb2.y, 3);
|
|
|
|
assertEquals(bb1.width, bb2.height, 3);
|
|
|
|
assertEquals(bb1.width, bb2.height, 3);
|
|
|
|
}
|
2018-02-23 17:04:28 -05:00
|
|
|
}
|