Commit 017cc770 authored by Sanford Jonathan Edelist's avatar Sanford Jonathan Edelist
Browse files

Update YOLOTest.java

parent fe0c40bf

src/test/java/yolo/YOLOTest.java

+6 −6
Original line number Diff line number Diff line
@@ -15,18 +15,18 @@ public class YOLOTest { 
    private static YOLONet yoloNet;



    public static void main(String[] args) {

        String imagePath = "src/main/java/yolo/TestFrame.png";

        String outputDir = "src/main/java/yolo";

        String csvDir = "src/main/java/yolo";

        String imagePath = "src/test/java/yolo/TestFrame.png";

        String outputDir = "src/test/java/yolo";

        String csvDir = "src/test/java/yolo";



        // Ensure output directories exist

        new File(outputDir).mkdirs();

        new File(csvDir).mkdirs();



        yoloNet = new YOLONet(

            "src/main/java/yolo/yolov4.cfg",

            "src/main/java/yolo/yolov4.weights",

            "src/main/java/yolo/coco.names",

            "src/test/java/yolo/yolov4.cfg",

            "src/test/java/yolo/yolov4.weights",

            "src/test/java/yolo/coco.names",

                608, 608);



        if (!yoloNet.setup()) {