initial commit ; currently it wont build :C
This commit is contained in:
43
.gitignore
vendored
Normal file
43
.gitignore
vendored
Normal file
@@ -0,0 +1,43 @@
|
|||||||
|
.gradle
|
||||||
|
build/
|
||||||
|
!gradle/wrapper/gradle-wrapper.jar
|
||||||
|
!**/src/main/**/build/
|
||||||
|
!**/src/test/**/build/
|
||||||
|
|
||||||
|
### IntelliJ IDEA ###
|
||||||
|
.idea/modules.xml
|
||||||
|
.idea/jarRepositories.xml
|
||||||
|
.idea/compiler.xml
|
||||||
|
.idea/libraries/
|
||||||
|
*.iws
|
||||||
|
*.iml
|
||||||
|
*.ipr
|
||||||
|
out/
|
||||||
|
!**/src/main/**/out/
|
||||||
|
!**/src/test/**/out/
|
||||||
|
|
||||||
|
### Eclipse ###
|
||||||
|
.apt_generated
|
||||||
|
.classpath
|
||||||
|
.factorypath
|
||||||
|
.project
|
||||||
|
.settings
|
||||||
|
.springBeans
|
||||||
|
.sts4-cache
|
||||||
|
bin/
|
||||||
|
!**/src/main/**/bin/
|
||||||
|
!**/src/test/**/bin/
|
||||||
|
|
||||||
|
### NetBeans ###
|
||||||
|
/nbproject/private/
|
||||||
|
/nbbuild/
|
||||||
|
/dist/
|
||||||
|
/nbdist/
|
||||||
|
/.nb-gradle/
|
||||||
|
|
||||||
|
### VS Code ###
|
||||||
|
.vscode/
|
||||||
|
|
||||||
|
### Mac OS ###
|
||||||
|
.DS_Store
|
||||||
|
/.idea/
|
||||||
20
build.gradle.kts
Normal file
20
build.gradle.kts
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
plugins {
|
||||||
|
id("java")
|
||||||
|
}
|
||||||
|
|
||||||
|
group = "org.jdetect"
|
||||||
|
version = "1.0-SNAPSHOT"
|
||||||
|
|
||||||
|
repositories {
|
||||||
|
mavenCentral()
|
||||||
|
}
|
||||||
|
|
||||||
|
dependencies {
|
||||||
|
implementation("org.openpnp:opencv:4.5.1-2")
|
||||||
|
testImplementation(platform("org.junit:junit-bom:5.10.0"))
|
||||||
|
testImplementation("org.junit.jupiter:junit-jupiter")
|
||||||
|
}
|
||||||
|
|
||||||
|
tasks.test {
|
||||||
|
useJUnitPlatform()
|
||||||
|
}
|
||||||
BIN
gradle/wrapper/gradle-wrapper.jar
vendored
Normal file
BIN
gradle/wrapper/gradle-wrapper.jar
vendored
Normal file
Binary file not shown.
6
gradle/wrapper/gradle-wrapper.properties
vendored
Normal file
6
gradle/wrapper/gradle-wrapper.properties
vendored
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
#Sun May 25 13:46:09 CEST 2025
|
||||||
|
distributionBase=GRADLE_USER_HOME
|
||||||
|
distributionPath=wrapper/dists
|
||||||
|
distributionUrl=https\://services.gradle.org/distributions/gradle-8.14-bin.zip
|
||||||
|
zipStoreBase=GRADLE_USER_HOME
|
||||||
|
zipStorePath=wrapper/dists
|
||||||
234
gradlew
vendored
Executable file
234
gradlew
vendored
Executable file
@@ -0,0 +1,234 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
#
|
||||||
|
# Copyright © 2015-2021 the original authors.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
#
|
||||||
|
|
||||||
|
##############################################################################
|
||||||
|
#
|
||||||
|
# Gradle start up script for POSIX generated by Gradle.
|
||||||
|
#
|
||||||
|
# Important for running:
|
||||||
|
#
|
||||||
|
# (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is
|
||||||
|
# noncompliant, but you have some other compliant shell such as ksh or
|
||||||
|
# bash, then to run this script, type that shell name before the whole
|
||||||
|
# command line, like:
|
||||||
|
#
|
||||||
|
# ksh Gradle
|
||||||
|
#
|
||||||
|
# Busybox and similar reduced shells will NOT work, because this script
|
||||||
|
# requires all of these POSIX shell features:
|
||||||
|
# * functions;
|
||||||
|
# * expansions «$var», «${var}», «${var:-default}», «${var+SET}»,
|
||||||
|
# «${var#prefix}», «${var%suffix}», and «$( cmd )»;
|
||||||
|
# * compound commands having a testable exit status, especially «case»;
|
||||||
|
# * various built-in commands including «command», «set», and «ulimit».
|
||||||
|
#
|
||||||
|
# Important for patching:
|
||||||
|
#
|
||||||
|
# (2) This script targets any POSIX shell, so it avoids extensions provided
|
||||||
|
# by Bash, Ksh, etc; in particular arrays are avoided.
|
||||||
|
#
|
||||||
|
# The "traditional" practice of packing multiple parameters into a
|
||||||
|
# space-separated string is a well documented source of bugs and security
|
||||||
|
# problems, so this is (mostly) avoided, by progressively accumulating
|
||||||
|
# options in "$@", and eventually passing that to Java.
|
||||||
|
#
|
||||||
|
# Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS,
|
||||||
|
# and GRADLE_OPTS) rely on word-splitting, this is performed explicitly;
|
||||||
|
# see the in-line comments for details.
|
||||||
|
#
|
||||||
|
# There are tweaks for specific operating systems such as AIX, CygWin,
|
||||||
|
# Darwin, MinGW, and NonStop.
|
||||||
|
#
|
||||||
|
# (3) This script is generated from the Groovy template
|
||||||
|
# https://github.com/gradle/gradle/blob/master/subprojects/plugins/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt
|
||||||
|
# within the Gradle project.
|
||||||
|
#
|
||||||
|
# You can find Gradle at https://github.com/gradle/gradle/.
|
||||||
|
#
|
||||||
|
##############################################################################
|
||||||
|
|
||||||
|
# Attempt to set APP_HOME
|
||||||
|
|
||||||
|
# Resolve links: $0 may be a link
|
||||||
|
app_path=$0
|
||||||
|
|
||||||
|
# Need this for daisy-chained symlinks.
|
||||||
|
while
|
||||||
|
APP_HOME=${app_path%"${app_path##*/}"} # leaves a trailing /; empty if no leading path
|
||||||
|
[ -h "$app_path" ]
|
||||||
|
do
|
||||||
|
ls=$( ls -ld "$app_path" )
|
||||||
|
link=${ls#*' -> '}
|
||||||
|
case $link in #(
|
||||||
|
/*) app_path=$link ;; #(
|
||||||
|
*) app_path=$APP_HOME$link ;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
APP_HOME=$( cd "${APP_HOME:-./}" && pwd -P ) || exit
|
||||||
|
|
||||||
|
APP_NAME="Gradle"
|
||||||
|
APP_BASE_NAME=${0##*/}
|
||||||
|
|
||||||
|
# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
|
||||||
|
DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"'
|
||||||
|
|
||||||
|
# Use the maximum available, or set MAX_FD != -1 to use that value.
|
||||||
|
MAX_FD=maximum
|
||||||
|
|
||||||
|
warn () {
|
||||||
|
echo "$*"
|
||||||
|
} >&2
|
||||||
|
|
||||||
|
die () {
|
||||||
|
echo
|
||||||
|
echo "$*"
|
||||||
|
echo
|
||||||
|
exit 1
|
||||||
|
} >&2
|
||||||
|
|
||||||
|
# OS specific support (must be 'true' or 'false').
|
||||||
|
cygwin=false
|
||||||
|
msys=false
|
||||||
|
darwin=false
|
||||||
|
nonstop=false
|
||||||
|
case "$( uname )" in #(
|
||||||
|
CYGWIN* ) cygwin=true ;; #(
|
||||||
|
Darwin* ) darwin=true ;; #(
|
||||||
|
MSYS* | MINGW* ) msys=true ;; #(
|
||||||
|
NONSTOP* ) nonstop=true ;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
|
||||||
|
|
||||||
|
|
||||||
|
# Determine the Java command to use to start the JVM.
|
||||||
|
if [ -n "$JAVA_HOME" ] ; then
|
||||||
|
if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
|
||||||
|
# IBM's JDK on AIX uses strange locations for the executables
|
||||||
|
JAVACMD=$JAVA_HOME/jre/sh/java
|
||||||
|
else
|
||||||
|
JAVACMD=$JAVA_HOME/bin/java
|
||||||
|
fi
|
||||||
|
if [ ! -x "$JAVACMD" ] ; then
|
||||||
|
die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
|
||||||
|
|
||||||
|
Please set the JAVA_HOME variable in your environment to match the
|
||||||
|
location of your Java installation."
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
JAVACMD=java
|
||||||
|
which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
|
||||||
|
|
||||||
|
Please set the JAVA_HOME variable in your environment to match the
|
||||||
|
location of your Java installation."
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Increase the maximum file descriptors if we can.
|
||||||
|
if ! "$cygwin" && ! "$darwin" && ! "$nonstop" ; then
|
||||||
|
case $MAX_FD in #(
|
||||||
|
max*)
|
||||||
|
MAX_FD=$( ulimit -H -n ) ||
|
||||||
|
warn "Could not query maximum file descriptor limit"
|
||||||
|
esac
|
||||||
|
case $MAX_FD in #(
|
||||||
|
'' | soft) :;; #(
|
||||||
|
*)
|
||||||
|
ulimit -n "$MAX_FD" ||
|
||||||
|
warn "Could not set maximum file descriptor limit to $MAX_FD"
|
||||||
|
esac
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Collect all arguments for the java command, stacking in reverse order:
|
||||||
|
# * args from the command line
|
||||||
|
# * the main class name
|
||||||
|
# * -classpath
|
||||||
|
# * -D...appname settings
|
||||||
|
# * --module-path (only if needed)
|
||||||
|
# * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables.
|
||||||
|
|
||||||
|
# For Cygwin or MSYS, switch paths to Windows format before running java
|
||||||
|
if "$cygwin" || "$msys" ; then
|
||||||
|
APP_HOME=$( cygpath --path --mixed "$APP_HOME" )
|
||||||
|
CLASSPATH=$( cygpath --path --mixed "$CLASSPATH" )
|
||||||
|
|
||||||
|
JAVACMD=$( cygpath --unix "$JAVACMD" )
|
||||||
|
|
||||||
|
# Now convert the arguments - kludge to limit ourselves to /bin/sh
|
||||||
|
for arg do
|
||||||
|
if
|
||||||
|
case $arg in #(
|
||||||
|
-*) false ;; # don't mess with options #(
|
||||||
|
/?*) t=${arg#/} t=/${t%%/*} # looks like a POSIX filepath
|
||||||
|
[ -e "$t" ] ;; #(
|
||||||
|
*) false ;;
|
||||||
|
esac
|
||||||
|
then
|
||||||
|
arg=$( cygpath --path --ignore --mixed "$arg" )
|
||||||
|
fi
|
||||||
|
# Roll the args list around exactly as many times as the number of
|
||||||
|
# args, so each arg winds up back in the position where it started, but
|
||||||
|
# possibly modified.
|
||||||
|
#
|
||||||
|
# NB: a `for` loop captures its iteration list before it begins, so
|
||||||
|
# changing the positional parameters here affects neither the number of
|
||||||
|
# iterations, nor the values presented in `arg`.
|
||||||
|
shift # remove old arg
|
||||||
|
set -- "$@" "$arg" # push replacement arg
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Collect all arguments for the java command;
|
||||||
|
# * $DEFAULT_JVM_OPTS, $JAVA_OPTS, and $GRADLE_OPTS can contain fragments of
|
||||||
|
# shell script including quotes and variable substitutions, so put them in
|
||||||
|
# double quotes to make sure that they get re-expanded; and
|
||||||
|
# * put everything else in single quotes, so that it's not re-expanded.
|
||||||
|
|
||||||
|
set -- \
|
||||||
|
"-Dorg.gradle.appname=$APP_BASE_NAME" \
|
||||||
|
-classpath "$CLASSPATH" \
|
||||||
|
org.gradle.wrapper.GradleWrapperMain \
|
||||||
|
"$@"
|
||||||
|
|
||||||
|
# Use "xargs" to parse quoted args.
|
||||||
|
#
|
||||||
|
# With -n1 it outputs one arg per line, with the quotes and backslashes removed.
|
||||||
|
#
|
||||||
|
# In Bash we could simply go:
|
||||||
|
#
|
||||||
|
# readarray ARGS < <( xargs -n1 <<<"$var" ) &&
|
||||||
|
# set -- "${ARGS[@]}" "$@"
|
||||||
|
#
|
||||||
|
# but POSIX shell has neither arrays nor command substitution, so instead we
|
||||||
|
# post-process each arg (as a line of input to sed) to backslash-escape any
|
||||||
|
# character that might be a shell metacharacter, then use eval to reverse
|
||||||
|
# that process (while maintaining the separation between arguments), and wrap
|
||||||
|
# the whole thing up as a single "set" statement.
|
||||||
|
#
|
||||||
|
# This will of course break if any of these variables contains a newline or
|
||||||
|
# an unmatched quote.
|
||||||
|
#
|
||||||
|
|
||||||
|
eval "set -- $(
|
||||||
|
printf '%s\n' "$DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS" |
|
||||||
|
xargs -n1 |
|
||||||
|
sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' |
|
||||||
|
tr '\n' ' '
|
||||||
|
)" '"$@"'
|
||||||
|
|
||||||
|
exec "$JAVACMD" "$@"
|
||||||
89
gradlew.bat
vendored
Normal file
89
gradlew.bat
vendored
Normal file
@@ -0,0 +1,89 @@
|
|||||||
|
@rem
|
||||||
|
@rem Copyright 2015 the original author or authors.
|
||||||
|
@rem
|
||||||
|
@rem Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
@rem you may not use this file except in compliance with the License.
|
||||||
|
@rem You may obtain a copy of the License at
|
||||||
|
@rem
|
||||||
|
@rem https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
@rem
|
||||||
|
@rem Unless required by applicable law or agreed to in writing, software
|
||||||
|
@rem distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
@rem See the License for the specific language governing permissions and
|
||||||
|
@rem limitations under the License.
|
||||||
|
@rem
|
||||||
|
|
||||||
|
@if "%DEBUG%" == "" @echo off
|
||||||
|
@rem ##########################################################################
|
||||||
|
@rem
|
||||||
|
@rem Gradle startup script for Windows
|
||||||
|
@rem
|
||||||
|
@rem ##########################################################################
|
||||||
|
|
||||||
|
@rem Set local scope for the variables with windows NT shell
|
||||||
|
if "%OS%"=="Windows_NT" setlocal
|
||||||
|
|
||||||
|
set DIRNAME=%~dp0
|
||||||
|
if "%DIRNAME%" == "" set DIRNAME=.
|
||||||
|
set APP_BASE_NAME=%~n0
|
||||||
|
set APP_HOME=%DIRNAME%
|
||||||
|
|
||||||
|
@rem Resolve any "." and ".." in APP_HOME to make it shorter.
|
||||||
|
for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi
|
||||||
|
|
||||||
|
@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
|
||||||
|
set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m"
|
||||||
|
|
||||||
|
@rem Find java.exe
|
||||||
|
if defined JAVA_HOME goto findJavaFromJavaHome
|
||||||
|
|
||||||
|
set JAVA_EXE=java.exe
|
||||||
|
%JAVA_EXE% -version >NUL 2>&1
|
||||||
|
if "%ERRORLEVEL%" == "0" goto execute
|
||||||
|
|
||||||
|
echo.
|
||||||
|
echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
|
||||||
|
echo.
|
||||||
|
echo Please set the JAVA_HOME variable in your environment to match the
|
||||||
|
echo location of your Java installation.
|
||||||
|
|
||||||
|
goto fail
|
||||||
|
|
||||||
|
:findJavaFromJavaHome
|
||||||
|
set JAVA_HOME=%JAVA_HOME:"=%
|
||||||
|
set JAVA_EXE=%JAVA_HOME%/bin/java.exe
|
||||||
|
|
||||||
|
if exist "%JAVA_EXE%" goto execute
|
||||||
|
|
||||||
|
echo.
|
||||||
|
echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
|
||||||
|
echo.
|
||||||
|
echo Please set the JAVA_HOME variable in your environment to match the
|
||||||
|
echo location of your Java installation.
|
||||||
|
|
||||||
|
goto fail
|
||||||
|
|
||||||
|
:execute
|
||||||
|
@rem Setup the command line
|
||||||
|
|
||||||
|
set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
|
||||||
|
|
||||||
|
|
||||||
|
@rem Execute Gradle
|
||||||
|
"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %*
|
||||||
|
|
||||||
|
:end
|
||||||
|
@rem End local scope for the variables with windows NT shell
|
||||||
|
if "%ERRORLEVEL%"=="0" goto mainEnd
|
||||||
|
|
||||||
|
:fail
|
||||||
|
rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
|
||||||
|
rem the _cmd.exe /c_ return code!
|
||||||
|
if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
|
||||||
|
exit /b 1
|
||||||
|
|
||||||
|
:mainEnd
|
||||||
|
if "%OS%"=="Windows_NT" endlocal
|
||||||
|
|
||||||
|
:omega
|
||||||
1
settings.gradle.kts
Normal file
1
settings.gradle.kts
Normal file
@@ -0,0 +1 @@
|
|||||||
|
rootProject.name = "MultiJDetect"
|
||||||
126
src/main/java/org/jdetect/CameraPanel.java
Normal file
126
src/main/java/org/jdetect/CameraPanel.java
Normal file
@@ -0,0 +1,126 @@
|
|||||||
|
package org.jdetect;
|
||||||
|
|
||||||
|
import org.opencv.core.*;
|
||||||
|
import org.opencv.videoio.VideoCapture;
|
||||||
|
|
||||||
|
import javax.swing.*;
|
||||||
|
import java.awt.*;
|
||||||
|
import java.awt.event.ActionEvent;
|
||||||
|
import java.awt.image.BufferedImage;
|
||||||
|
|
||||||
|
public class CameraPanel extends JPanel {
|
||||||
|
private final VideoCapture cap;
|
||||||
|
private final JLabel videoLabel;
|
||||||
|
private final JLabel statusLabel;
|
||||||
|
private DetectionThread detectionThread;
|
||||||
|
private Timer timer;
|
||||||
|
private final String source;
|
||||||
|
private int frameCounter = 0;
|
||||||
|
private static final int DETECTION_INTERVAL = 5;
|
||||||
|
|
||||||
|
public CameraPanel(String source) {
|
||||||
|
this.source = source;
|
||||||
|
boolean isStream = source.startsWith("rtsp://") || source.startsWith("https://");
|
||||||
|
|
||||||
|
setLayout(new BorderLayout());
|
||||||
|
setBorder(BorderFactory.createEtchedBorder());
|
||||||
|
|
||||||
|
// Video display
|
||||||
|
videoLabel = new JLabel();
|
||||||
|
videoLabel.setHorizontalAlignment(SwingConstants.CENTER);
|
||||||
|
add(videoLabel, BorderLayout.CENTER);
|
||||||
|
|
||||||
|
// Controls panel
|
||||||
|
JPanel controlsPanel = new JPanel(new FlowLayout(FlowLayout.LEFT));
|
||||||
|
|
||||||
|
JButton snapshotButton = new JButton("Snapshot");
|
||||||
|
snapshotButton.addActionListener(this::takeSnapshot);
|
||||||
|
controlsPanel.add(snapshotButton);
|
||||||
|
|
||||||
|
JButton stopButton = new JButton("Remove");
|
||||||
|
stopButton.addActionListener(e -> stop());
|
||||||
|
controlsPanel.add(stopButton);
|
||||||
|
|
||||||
|
add(controlsPanel, BorderLayout.SOUTH);
|
||||||
|
|
||||||
|
// Status label
|
||||||
|
statusLabel = new JLabel("Initializing...");
|
||||||
|
add(statusLabel, BorderLayout.NORTH);
|
||||||
|
|
||||||
|
// Initialize video source
|
||||||
|
if (isStream) {
|
||||||
|
cap = new VideoCapture(source);
|
||||||
|
} else {
|
||||||
|
try {
|
||||||
|
int index = Integer.parseInt(source);
|
||||||
|
cap = new VideoCapture(index);
|
||||||
|
} catch (NumberFormatException e) {
|
||||||
|
throw new IllegalArgumentException("Invalid camera source: " + source);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!cap.isOpened()) {
|
||||||
|
statusLabel.setText("Failed to open: " + source);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
timer = new Timer(30, e -> updateFrame());
|
||||||
|
timer.start();
|
||||||
|
statusLabel.setText("Active: " + source);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void updateFrame() {
|
||||||
|
Mat frame = new Mat();
|
||||||
|
if (!cap.read(frame)) {
|
||||||
|
statusLabel.setText("Error reading frame from: " + source);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Process every Nth frame for detection
|
||||||
|
if (frameCounter++ % DETECTION_INTERVAL == 0 && ModelLoader.getNet() != null) {
|
||||||
|
if (detectionThread == null || detectionThread.isDone()) {
|
||||||
|
Mat frameCopy = frame.clone();
|
||||||
|
detectionThread = new DetectionThread(
|
||||||
|
ModelLoader.getNet(),
|
||||||
|
ModelLoader.getClasses(),
|
||||||
|
ModelLoader.getOutputLayers(),
|
||||||
|
frameCopy,
|
||||||
|
this::updateWithDetections
|
||||||
|
);
|
||||||
|
detectionThread.execute();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
displayFrame(frame);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void updateWithDetections(Mat frame) {
|
||||||
|
SwingUtilities.invokeLater(() -> displayFrame(frame));
|
||||||
|
}
|
||||||
|
|
||||||
|
private void displayFrame(Mat frame) {
|
||||||
|
BufferedImage image = ImageUtils.matToBufferedImage(frame);
|
||||||
|
ImageIcon icon = new ImageIcon(image);
|
||||||
|
videoLabel.setIcon(icon);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void takeSnapshot(ActionEvent e) {
|
||||||
|
Mat frame = new Mat();
|
||||||
|
if (cap.read(frame) && !frame.empty()) {
|
||||||
|
ImageUtils.saveImage(frame);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void stop() {
|
||||||
|
timer.stop();
|
||||||
|
if (cap != null) cap.release();
|
||||||
|
if (detectionThread != null) detectionThread.cancel(true);
|
||||||
|
|
||||||
|
Container parent = getParent();
|
||||||
|
if (parent != null) {
|
||||||
|
parent.remove(this);
|
||||||
|
parent.revalidate();
|
||||||
|
parent.repaint();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
127
src/main/java/org/jdetect/DetectionThread.java
Normal file
127
src/main/java/org/jdetect/DetectionThread.java
Normal file
@@ -0,0 +1,127 @@
|
|||||||
|
package org.jdetect;
|
||||||
|
|
||||||
|
import org.opencv.core.*;
|
||||||
|
import org.opencv.dnn.Dnn;
|
||||||
|
import org.opencv.imgproc.Imgproc;
|
||||||
|
import org.opencv.dnn.Net;
|
||||||
|
|
||||||
|
import javax.swing.*;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
public class DetectionThread extends SwingWorker<Mat, Void> {
|
||||||
|
private final Net net;
|
||||||
|
private final List<String> classes;
|
||||||
|
private final List<String> outputLayers;
|
||||||
|
private final Mat frame;
|
||||||
|
private final DetectionCallback callback;
|
||||||
|
|
||||||
|
public DetectionThread(Net net, List<String> classes, List<String> outputLayers,
|
||||||
|
Mat frame, DetectionCallback callback) {
|
||||||
|
this.net = net;
|
||||||
|
this.classes = classes;
|
||||||
|
this.outputLayers = outputLayers;
|
||||||
|
this.frame = frame;
|
||||||
|
this.callback = callback;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected Mat doInBackground() {
|
||||||
|
if (net == null || classes == null) return frame;
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Create blob from image (resize to 416x416 for YOLO)
|
||||||
|
Mat blob = Dnn.blobFromImage(frame, 1.0/255.0, new Size(416, 416),
|
||||||
|
new Scalar(0), true, false);
|
||||||
|
|
||||||
|
net.setInput(blob);
|
||||||
|
List<Mat> outs = new ArrayList<>();
|
||||||
|
net.forward(outs, outputLayers);
|
||||||
|
|
||||||
|
// Process detections
|
||||||
|
processDetections(frame, outs);
|
||||||
|
return frame;
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
return frame;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void processDetections(Mat frame, List<Mat> outs) {
|
||||||
|
List<Rect> boxes = new ArrayList<>();
|
||||||
|
List<Float> confidences = new ArrayList<>();
|
||||||
|
List<Integer> classIds = new ArrayList<>();
|
||||||
|
|
||||||
|
for (Mat out : outs) {
|
||||||
|
float[] data = new float[(int) out.total() * out.channels()];
|
||||||
|
out.get(0, 0, data);
|
||||||
|
|
||||||
|
for (int i = 0; i < out.rows(); i++) {
|
||||||
|
int offset = i * out.cols();
|
||||||
|
float confidence = data[offset + 4];
|
||||||
|
|
||||||
|
if (confidence > 0.5) { // Confidence threshold
|
||||||
|
int classId = 0;
|
||||||
|
float maxScore = 0;
|
||||||
|
for (int j = 5; j < out.cols(); j++) {
|
||||||
|
if (data[offset + j] > maxScore) {
|
||||||
|
maxScore = data[offset + j];
|
||||||
|
classId = j - 5;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (maxScore > 0.5 && classId < classes.size()) {
|
||||||
|
float width = data[offset + 2] * frame.cols();
|
||||||
|
float height = data[offset + 3] * frame.rows();
|
||||||
|
float x = (data[offset] * frame.cols()) - (width / 2);
|
||||||
|
float y = (data[offset + 1] * frame.rows()) - (height / 2);
|
||||||
|
|
||||||
|
boxes.add(new Rect((int)x, (int)y, (int)width, (int)height));
|
||||||
|
confidences.add(confidence);
|
||||||
|
classIds.add(classId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply Non-Maximum Suppression
|
||||||
|
List<Rect2d> rect2dBoxes = new ArrayList<>();
|
||||||
|
for (Rect box : boxes) {
|
||||||
|
rect2dBoxes.add(new Rect2d(box.x, box.y, box.width, box.height));
|
||||||
|
}
|
||||||
|
MatOfRect2d boxesMat = new MatOfRect2d();
|
||||||
|
boxesMat.fromList(rect2dBoxes);
|
||||||
|
|
||||||
|
MatOfFloat confidencesMat = new MatOfFloat();
|
||||||
|
confidencesMat.fromList(confidences);
|
||||||
|
MatOfInt indices = new MatOfInt();
|
||||||
|
Dnn.NMSBoxes(boxesMat, confidencesMat, 0.5f, 0.4f, indices);
|
||||||
|
|
||||||
|
// Draw detections
|
||||||
|
for (int i : indices.toArray()) {
|
||||||
|
Rect box = boxes.get(i);
|
||||||
|
String label = classes.get(classIds.get(i));
|
||||||
|
float confidence = confidences.get(i);
|
||||||
|
|
||||||
|
Imgproc.rectangle(frame, box, new Scalar(0, 255, 0), 2);
|
||||||
|
String text = String.format("%s %.2f", label, confidence);
|
||||||
|
Imgproc.putText(frame, text, new Point(box.x, box.y - 5),
|
||||||
|
Imgproc.FONT_HERSHEY_SIMPLEX, 0.5, new Scalar(0, 255, 0), 1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void done() {
|
||||||
|
try {
|
||||||
|
if (!isCancelled()) {
|
||||||
|
callback.onDetectionComplete(get());
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public interface DetectionCallback {
|
||||||
|
void onDetectionComplete(Mat frame);
|
||||||
|
}
|
||||||
|
}
|
||||||
68
src/main/java/org/jdetect/ImageUtils.java
Normal file
68
src/main/java/org/jdetect/ImageUtils.java
Normal file
@@ -0,0 +1,68 @@
|
|||||||
|
package org.jdetect;
|
||||||
|
|
||||||
|
import org.opencv.core.*;
|
||||||
|
import org.opencv.imgcodecs.Imgcodecs;
|
||||||
|
import org.opencv.imgproc.Imgproc;
|
||||||
|
|
||||||
|
import javax.swing.*;
|
||||||
|
import java.awt.image.BufferedImage;
|
||||||
|
import java.awt.image.DataBufferByte;
|
||||||
|
import java.io.File;
|
||||||
|
import java.text.SimpleDateFormat;
|
||||||
|
import java.util.Date;
|
||||||
|
|
||||||
|
public class ImageUtils {
|
||||||
|
public static BufferedImage matToBufferedImage(Mat mat) {
|
||||||
|
if (mat.channels() == 1) {
|
||||||
|
return grayMatToBufferedImage(mat);
|
||||||
|
} else {
|
||||||
|
return colorMatToBufferedImage(mat);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static BufferedImage grayMatToBufferedImage(Mat mat) {
|
||||||
|
BufferedImage image = new BufferedImage(mat.cols(), mat.rows(), BufferedImage.TYPE_BYTE_GRAY);
|
||||||
|
byte[] data = ((DataBufferByte) image.getRaster().getDataBuffer()).getData();
|
||||||
|
mat.get(0, 0, data);
|
||||||
|
return image;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static BufferedImage colorMatToBufferedImage(Mat mat) {
|
||||||
|
Mat rgb = new Mat();
|
||||||
|
Imgproc.cvtColor(mat, rgb, Imgproc.COLOR_BGR2RGB);
|
||||||
|
|
||||||
|
BufferedImage image = new BufferedImage(rgb.cols(), rgb.rows(), BufferedImage.TYPE_3BYTE_BGR);
|
||||||
|
byte[] data = ((DataBufferByte) image.getRaster().getDataBuffer()).getData();
|
||||||
|
rgb.get(0, 0, data);
|
||||||
|
return image;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void saveImage(Mat frame) {
|
||||||
|
JFileChooser fileChooser = new JFileChooser();
|
||||||
|
fileChooser.setDialogTitle("Save Snapshot");
|
||||||
|
|
||||||
|
// Suggest filename with timestamp
|
||||||
|
String timestamp = new SimpleDateFormat("yyyyMMdd-HHmmss").format(new Date());
|
||||||
|
fileChooser.setSelectedFile(new File("snapshot_" + timestamp + ".jpg"));
|
||||||
|
|
||||||
|
if (fileChooser.showSaveDialog(null) == JFileChooser.APPROVE_OPTION) {
|
||||||
|
File file = fileChooser.getSelectedFile();
|
||||||
|
String filename = file.toString();
|
||||||
|
|
||||||
|
// Ensure extension
|
||||||
|
if (!filename.toLowerCase().endsWith(".jpg") && !filename.toLowerCase().endsWith(".png")) {
|
||||||
|
filename += ".jpg";
|
||||||
|
}
|
||||||
|
|
||||||
|
// Convert to RGB before saving
|
||||||
|
Mat rgb = new Mat();
|
||||||
|
Imgproc.cvtColor(frame, rgb, Imgproc.COLOR_BGR2RGB);
|
||||||
|
|
||||||
|
if (Imgcodecs.imwrite(filename, rgb)) {
|
||||||
|
JOptionPane.showMessageDialog(null, "Image saved successfully!");
|
||||||
|
} else {
|
||||||
|
JOptionPane.showMessageDialog(null, "Failed to save image!", "Error", JOptionPane.ERROR_MESSAGE);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
127
src/main/java/org/jdetect/MainApp.java
Normal file
127
src/main/java/org/jdetect/MainApp.java
Normal file
@@ -0,0 +1,127 @@
|
|||||||
|
package org.jdetect;
|
||||||
|
|
||||||
|
import org.opencv.core.Core;
|
||||||
|
import javax.swing.*;
|
||||||
|
import java.awt.*;
|
||||||
|
import java.awt.event.ActionEvent;
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
|
public class MainApp extends JFrame {
|
||||||
|
private final JPanel camerasPanel;
|
||||||
|
|
||||||
|
static {
|
||||||
|
System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
|
||||||
|
}
|
||||||
|
|
||||||
|
public MainApp() {
|
||||||
|
setTitle("Multi-Camera Object Detection");
|
||||||
|
setDefaultCloseOperation(EXIT_ON_CLOSE);
|
||||||
|
setSize(1200, 800);
|
||||||
|
|
||||||
|
// Load model
|
||||||
|
try {
|
||||||
|
ModelLoader.loadModel();
|
||||||
|
} catch (IOException e) {
|
||||||
|
JOptionPane.showMessageDialog(this, "Failed to load model: " + e.getMessage(),
|
||||||
|
"Error", JOptionPane.ERROR_MESSAGE);
|
||||||
|
System.exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Main content
|
||||||
|
camerasPanel = new JPanel(new WrapLayout(FlowLayout.LEFT, 10, 10));
|
||||||
|
add(new JScrollPane(camerasPanel), BorderLayout.CENTER);
|
||||||
|
|
||||||
|
// Control panel
|
||||||
|
JPanel controlPanel = new JPanel();
|
||||||
|
|
||||||
|
JButton addCameraBtn = new JButton("Add Camera");
|
||||||
|
addCameraBtn.addActionListener(this::addCamera);
|
||||||
|
controlPanel.add(addCameraBtn);
|
||||||
|
|
||||||
|
JButton addStreamBtn = new JButton("Add Stream");
|
||||||
|
addStreamBtn.addActionListener(this::addStream);
|
||||||
|
controlPanel.add(addStreamBtn);
|
||||||
|
|
||||||
|
add(controlPanel, BorderLayout.SOUTH);
|
||||||
|
|
||||||
|
// Add default camera
|
||||||
|
addCameraPanel("0");
|
||||||
|
}
|
||||||
|
|
||||||
|
private void addCamera(ActionEvent e) {
|
||||||
|
String input = JOptionPane.showInputDialog(this, "Enter camera index (0, 1, etc.):", "0");
|
||||||
|
if (input != null && !input.trim().isEmpty()) {
|
||||||
|
addCameraPanel(input.trim());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void addStream(ActionEvent e) {
|
||||||
|
String input = JOptionPane.showInputDialog(this,
|
||||||
|
"Enter stream URL (rtsp:// or http://):",
|
||||||
|
"rtsp://example.com/stream");
|
||||||
|
if (input != null && !input.trim().isEmpty()) {
|
||||||
|
addCameraPanel(input.trim());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void addCameraPanel(String source) {
|
||||||
|
CameraPanel panel = new CameraPanel(source);
|
||||||
|
camerasPanel.add(panel);
|
||||||
|
camerasPanel.revalidate();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void main(String[] args) {
|
||||||
|
try {
|
||||||
|
SwingUtilities.invokeAndWait(() -> {
|
||||||
|
try {
|
||||||
|
UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
|
||||||
|
MainApp app = new MainApp();
|
||||||
|
app.setVisible(true);
|
||||||
|
});
|
||||||
|
} catch (Exception ex) {
|
||||||
|
ex.printStackTrace();
|
||||||
|
System.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper class for wrapping camera panels
|
||||||
|
class WrapLayout extends FlowLayout {
|
||||||
|
|
||||||
|
public WrapLayout(int align, int hgap, int vgap) {
|
||||||
|
super(align, hgap, vgap);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Dimension preferredLayoutSize(Container target) {
|
||||||
|
synchronized (target.getTreeLock()) {
|
||||||
|
int targetWidth = target.getSize().width;
|
||||||
|
if (targetWidth == 0) {
|
||||||
|
targetWidth = Integer.MAX_VALUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
int x = 0;
|
||||||
|
int y = 0;
|
||||||
|
int rowHeight = 0;
|
||||||
|
|
||||||
|
for (Component comp : target.getComponents()) {
|
||||||
|
if (comp.isVisible()) {
|
||||||
|
Dimension pref = comp.getPreferredSize();
|
||||||
|
if (x + pref.width > targetWidth) {
|
||||||
|
y += rowHeight + getVgap();
|
||||||
|
x = 0;
|
||||||
|
rowHeight = 0;
|
||||||
|
}
|
||||||
|
x += pref.width + getHgap();
|
||||||
|
rowHeight = Math.max(rowHeight, pref.height);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
y += rowHeight;
|
||||||
|
return new Dimension(targetWidth, y);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
77
src/main/java/org/jdetect/ModelLoader.java
Normal file
77
src/main/java/org/jdetect/ModelLoader.java
Normal file
@@ -0,0 +1,77 @@
|
|||||||
|
package org.jdetect;
|
||||||
|
|
||||||
|
import org.opencv.core.*;
|
||||||
|
import org.opencv.dnn.Dnn;
|
||||||
|
import org.opencv.dnn.Net;
|
||||||
|
|
||||||
|
import java.io.BufferedReader;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.io.InputStreamReader;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
import java.nio.file.Paths;
|
||||||
|
|
||||||
|
|
||||||
|
public class ModelLoader {
|
||||||
|
private static Net net;
|
||||||
|
private static List<String> classes;
|
||||||
|
private static List<String> outputLayers;
|
||||||
|
|
||||||
|
public static synchronized void loadModel() throws IOException {
|
||||||
|
if (net != null) return;
|
||||||
|
|
||||||
|
// Load network
|
||||||
|
InputStream weightsStream = ModelLoader.class.getResourceAsStream("/yolov4-tiny.weights");
|
||||||
|
InputStream configStream = ModelLoader.class.getResourceAsStream("/yolov4-tiny.cfg");
|
||||||
|
|
||||||
|
assert weightsStream != null;
|
||||||
|
byte[] weights = weightsStream.readAllBytes();
|
||||||
|
assert configStream != null;
|
||||||
|
byte[] config = configStream.readAllBytes();
|
||||||
|
|
||||||
|
// Create temp files
|
||||||
|
String weightsPath = "temp_weights.weights";
|
||||||
|
String configPath = "temp_config.cfg";
|
||||||
|
|
||||||
|
Files.write(Paths.get(weightsPath), weights);
|
||||||
|
Files.write(Paths.get(configPath), config);
|
||||||
|
|
||||||
|
net = Dnn.readNetFromDarknet(configPath, weightsPath);
|
||||||
|
|
||||||
|
// Load classes
|
||||||
|
classes = new ArrayList<>();
|
||||||
|
try (InputStream classesStream = ModelLoader.class.getResourceAsStream("/coco.names")) {
|
||||||
|
assert classesStream != null;
|
||||||
|
try (BufferedReader br = new BufferedReader(new InputStreamReader(classesStream))) {
|
||||||
|
String line;
|
||||||
|
while ((line = br.readLine()) != null) {
|
||||||
|
classes.add(line.trim());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get output layers
|
||||||
|
List<String> layerNames = net.getLayerNames();
|
||||||
|
MatOfInt unconnectedOutLayers = net.getUnconnectedOutLayers();
|
||||||
|
int[] indices = unconnectedOutLayers.toArray();
|
||||||
|
|
||||||
|
outputLayers = new ArrayList<>();
|
||||||
|
for (int idx : indices) {
|
||||||
|
outputLayers.add(layerNames.get(idx - 1));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static Net getNet() {
|
||||||
|
return net;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static List<String> getClasses() {
|
||||||
|
return classes;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static List<String> getOutputLayers() {
|
||||||
|
return outputLayers;
|
||||||
|
}
|
||||||
|
}
|
||||||
80
src/main/resources/coco.names
Normal file
80
src/main/resources/coco.names
Normal file
@@ -0,0 +1,80 @@
|
|||||||
|
person
|
||||||
|
bicycle
|
||||||
|
car
|
||||||
|
motorbike
|
||||||
|
aeroplane
|
||||||
|
bus
|
||||||
|
train
|
||||||
|
truck
|
||||||
|
boat
|
||||||
|
traffic light
|
||||||
|
fire hydrant
|
||||||
|
stop sign
|
||||||
|
parking meter
|
||||||
|
bench
|
||||||
|
bird
|
||||||
|
cat
|
||||||
|
dog
|
||||||
|
horse
|
||||||
|
sheep
|
||||||
|
cow
|
||||||
|
elephant
|
||||||
|
bear
|
||||||
|
zebra
|
||||||
|
giraffe
|
||||||
|
backpack
|
||||||
|
umbrella
|
||||||
|
handbag
|
||||||
|
tie
|
||||||
|
suitcase
|
||||||
|
frisbee
|
||||||
|
skis
|
||||||
|
snowboard
|
||||||
|
sports ball
|
||||||
|
kite
|
||||||
|
baseball bat
|
||||||
|
baseball glove
|
||||||
|
skateboard
|
||||||
|
surfboard
|
||||||
|
tennis racket
|
||||||
|
bottle
|
||||||
|
wine glass
|
||||||
|
cup
|
||||||
|
fork
|
||||||
|
knife
|
||||||
|
spoon
|
||||||
|
bowl
|
||||||
|
banana
|
||||||
|
apple
|
||||||
|
sandwich
|
||||||
|
orange
|
||||||
|
broccoli
|
||||||
|
carrot
|
||||||
|
hot dog
|
||||||
|
pizza
|
||||||
|
donut
|
||||||
|
cake
|
||||||
|
chair
|
||||||
|
sofa
|
||||||
|
pottedplant
|
||||||
|
bed
|
||||||
|
diningtable
|
||||||
|
toilet
|
||||||
|
tvmonitor
|
||||||
|
laptop
|
||||||
|
mouse
|
||||||
|
remote
|
||||||
|
keyboard
|
||||||
|
cell phone
|
||||||
|
microwave
|
||||||
|
oven
|
||||||
|
toaster
|
||||||
|
sink
|
||||||
|
refrigerator
|
||||||
|
book
|
||||||
|
clock
|
||||||
|
vase
|
||||||
|
scissors
|
||||||
|
teddy bear
|
||||||
|
hair drier
|
||||||
|
toothbrush
|
||||||
294
src/main/resources/yolov4-tiny.cfg
Normal file
294
src/main/resources/yolov4-tiny.cfg
Normal file
@@ -0,0 +1,294 @@
|
|||||||
|
[net]
|
||||||
|
# Testing
|
||||||
|
#batch=1
|
||||||
|
#subdivisions=1
|
||||||
|
# Training
|
||||||
|
batch=64
|
||||||
|
subdivisions=1
|
||||||
|
width=416
|
||||||
|
height=416
|
||||||
|
channels=3
|
||||||
|
momentum=0.9
|
||||||
|
decay=0.0005
|
||||||
|
angle=0
|
||||||
|
saturation = 1.5
|
||||||
|
exposure = 1.5
|
||||||
|
hue=.1
|
||||||
|
|
||||||
|
learning_rate=0.00261
|
||||||
|
burn_in=1000
|
||||||
|
|
||||||
|
max_batches = 2000200
|
||||||
|
policy=steps
|
||||||
|
steps=1600000,1800000
|
||||||
|
scales=.1,.1
|
||||||
|
|
||||||
|
|
||||||
|
#weights_reject_freq=1001
|
||||||
|
#ema_alpha=0.9998
|
||||||
|
#equidistant_point=1000
|
||||||
|
#num_sigmas_reject_badlabels=3
|
||||||
|
#badlabels_rejection_percentage=0.2
|
||||||
|
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=32
|
||||||
|
size=3
|
||||||
|
stride=2
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=64
|
||||||
|
size=3
|
||||||
|
stride=2
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=64
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers=-1
|
||||||
|
groups=2
|
||||||
|
group_id=1
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=32
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=32
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers = -1,-2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=64
|
||||||
|
size=1
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers = -6,-1
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=128
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers=-1
|
||||||
|
groups=2
|
||||||
|
group_id=1
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=64
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=64
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers = -1,-2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=128
|
||||||
|
size=1
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers = -6,-1
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=256
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers=-1
|
||||||
|
groups=2
|
||||||
|
group_id=1
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=128
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=128
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers = -1,-2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=256
|
||||||
|
size=1
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers = -6,-1
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=512
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
##################################
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=256
|
||||||
|
size=1
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=512
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
size=1
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
filters=255
|
||||||
|
activation=linear
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
[yolo]
|
||||||
|
mask = 3,4,5
|
||||||
|
anchors = 10,14, 23,27, 37,58, 81,82, 135,169, 344,319
|
||||||
|
classes=80
|
||||||
|
num=6
|
||||||
|
jitter=.3
|
||||||
|
scale_x_y = 1.05
|
||||||
|
cls_normalizer=1.0
|
||||||
|
iou_normalizer=0.07
|
||||||
|
iou_loss=ciou
|
||||||
|
ignore_thresh = .7
|
||||||
|
truth_thresh = 1
|
||||||
|
random=0
|
||||||
|
resize=1.5
|
||||||
|
nms_kind=greedynms
|
||||||
|
beta_nms=0.6
|
||||||
|
#new_coords=1
|
||||||
|
#scale_x_y = 2.0
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers = -4
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=128
|
||||||
|
size=1
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[upsample]
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[route]
|
||||||
|
layers = -1, 23
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
batch_normalize=1
|
||||||
|
filters=256
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
size=1
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
filters=255
|
||||||
|
activation=linear
|
||||||
|
|
||||||
|
[yolo]
|
||||||
|
mask = 1,2,3
|
||||||
|
anchors = 10,14, 23,27, 37,58, 81,82, 135,169, 344,319
|
||||||
|
classes=80
|
||||||
|
num=6
|
||||||
|
jitter=.3
|
||||||
|
scale_x_y = 1.05
|
||||||
|
cls_normalizer=1.0
|
||||||
|
iou_normalizer=0.07
|
||||||
|
iou_loss=ciou
|
||||||
|
ignore_thresh = .7
|
||||||
|
truth_thresh = 1
|
||||||
|
random=0
|
||||||
|
resize=1.5
|
||||||
|
nms_kind=greedynms
|
||||||
|
beta_nms=0.6
|
||||||
|
#new_coords=1
|
||||||
|
#scale_x_y = 2.0
|
||||||
BIN
src/main/resources/yolov4-tiny.weights
Normal file
BIN
src/main/resources/yolov4-tiny.weights
Normal file
Binary file not shown.
Reference in New Issue
Block a user