Compare commits

..

7 Commits
master ... read

Author SHA1 Message Date
p57492168 48c3f57645 readme修改 2022-06-19 21:13:26 +08:00
p57492168 02a53f91cc 注释删除 2022-06-19 21:11:41 +08:00
p57492168 d41725a0b0 注释删除 2022-06-18 22:55:47 +08:00
p57492168 991d45640d 注释删除 2022-06-18 22:50:02 +08:00
p57492168 3db7a815da 按钮内容修改 2022-06-18 22:45:03 +08:00
p57492168 3e7cf3d707 添加按钮 2022-06-18 22:41:01 +08:00
p57492168 c7ed7f15c1 修改readme 2022-06-18 22:34:56 +08:00
18 changed files with 76 additions and 140 deletions

3
.idea/.gitignore vendored Normal file
View File

@ -0,0 +1,3 @@
# Default ignored files
/shelf/
/workspace.xml

View File

@ -0,0 +1,12 @@
<?xml version="1.0" encoding="UTF-8"?>
<module type="PYTHON_MODULE" version="4">
<component name="NewModuleRootManager">
<content url="file://$MODULE_DIR$" />
<orderEntry type="jdk" jdkName="Python 3.7 (Focus)" jdkType="Python SDK" />
<orderEntry type="sourceFolder" forTests="false" />
</component>
<component name="PyDocumentationSettings">
<option name="format" value="PLAIN" />
<option name="myDocStringFormat" value="Plain" />
</component>
</module>

View File

@ -0,0 +1,13 @@
<component name="InspectionProjectProfileManager">
<profile version="1.0">
<option name="myName" value="Project Default" />
<inspection_tool class="PyUnresolvedReferencesInspection" enabled="true" level="WARNING" enabled_by_default="true">
<option name="ignoredIdentifiers">
<list>
<option value="runtest.ImageEditorPage" />
<option value="ImageEditorPage" />
</list>
</option>
</inspection_tool>
</profile>
</component>

View File

@ -0,0 +1,6 @@
<component name="InspectionProjectProfileManager">
<settings>
<option name="USE_PROJECT_PROFILE" value="false" />
<version value="1.0" />
</settings>
</component>

4
.idea/misc.xml Normal file
View File

@ -0,0 +1,4 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.7 (Focus)" project-jdk-type="Python SDK" />
</project>

8
.idea/modules.xml Normal file
View File

@ -0,0 +1,8 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ProjectModuleManager">
<modules>
<module fileurl="file://$PROJECT_DIR$/.idea/PulseFocusPlatform.iml" filepath="$PROJECT_DIR$/.idea/PulseFocusPlatform.iml" />
</modules>
</component>
</project>

6
.idea/vcs.xml Normal file
View File

@ -0,0 +1,6 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="VcsDirectoryMappings">
<mapping directory="$PROJECT_DIR$" vcs="Git" />
</component>
</project>

119
README.md
View File

@ -6,14 +6,13 @@ Pulse Focus Platform脉冲聚焦是面向水底物体图像识别的实时检测
软件界面简单,易学易用,包含参数的输入选择,程序的运行,算法结果的展示等,源代码公开,算法可修改。 软件界面简单,易学易用,包含参数的输入选择,程序的运行,算法结果的展示等,源代码公开,算法可修改。
开发人员K. Wang、H.P. Yu、J. Li、Z.Y. Zhao、L.F. Zhang、G. Chen、H.T. Li、Z.Q. Wang、Y.G. Han 开发人员:王凯、于化鹏、李晶、王兆琦、李慧涛、赵志允、张乐飞、陈光
## 1. 开发环境配置 ## 1. 开发环境配置
运行以下命令: 运行以下命令:
```bash ```bash
conda env create -f create_env.yaml conda env create -f create_env.yaml
``` ```
该命令会创建一个名为`Focus`的conda虚拟环境用`conda activate Focus`即可激活该虚拟环境。 该命令会根据create_env.yaml配置文件创建一个名为`Focus`的conda虚拟环境用`conda activate Focus`即可激活该虚拟环境。
## 2. 软件运行 ## 2. 软件运行
@ -22,119 +21,11 @@ conda env create -f create_env.yaml
python main.py python main.py
``` ```
## 3. 软硬件运行平台 ## 3. 一些说明
1. 使用GPU版本
1配置要求
<table>
<tr>
<th>组件</th>
<th>配置</th>
<th>备注</th>
</tr>
<tr>
<td>系统 </td>
<td>Windows 10 家庭中文版 20H2 64位</td>
<td>扩展支持Linux和Mac系统</td>
</tr>
<tr>
<td>处理器</td>
<td>处理器类型:
酷睿i3兼容处理器或速度更快的处理器
处理器速度:
最低1.0GHz
建议2.0GHz或更快
</td>
<td>不支持ARM、IA64等芯片处理器</td>
</tr>
<tr>
<td>内存</td>
<td>RAM 16.0 GB (15.7 GB 可用)</td>
<td></td>
</tr>
<tr>
<td>显卡</td>
<td>最小:核心显卡
推荐GTX1060或同类型显卡
</td>
<td></td>
</tr>
<tr>
<td>硬盘</td>
<td>500G</td>
<td></td>
</tr>
<td>显示器</td>
<td>3840×2160像素高分屏</td>
<td></td>
</tr>
<tr>
</tr>
<td>软件</td>
<td>Anaconda3 2020及以上</td>
<td>Python3.7及以上,需手动安装包</td>
</tr>
</table>
2手动部署及运行
推荐的安装步骤如下:
安装Anaconda3-2020.02-Windows-x86_64或以上版本
手动安装pygame、pymunk、pyyaml、numpy、easydict和pyqt安装方式推荐参考如下
```
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pygame==2.0.1
```
将软件模块文件夹拷贝到电脑中以D盘为例路径为D:\island-multi_ships
## 4. 软件详细介绍
软件总体开发系统架构图如下所示。
![开发系统架构图](https://osredm.com/repo/PulseFocusPlatform/PulseFocusPlatform/raw/branch/master/pic1/1.png)
1界面设计
平台界面设计如上图所示,界面各组件功能设计如下:
![界面设计](https://osredm.com/repo/PulseFocusPlatform/PulseFocusPlatform/raw/branch/master/pic1/2.png)
* 静态图像导入用于选择需要进行预测的单张图像可支持jpgpngjpeg等格式图像选择图像后会在下方界面进行展示。
* 动态图像导入:用于选择需要进行预测的单个视频,可支持pm4等格式视频选择视频后会在下方界面进行展示。
* 信息导出:用于在预测完成后,将预测后的照片,视频导出到具体文件夹下。
* 特征选择:由于挑选相关特征。
* 预处理方法:由于选择相关预处理方法。
* 识别算法用于选择预测时的所需算法目前支持YOLO与RCNN两种模型算法。
* GPU加速选择是否使用GPU进行预测加速对视频预测加速效果明显。
* 识别:当相关配置完成后,点击识别选项,会进行预测处理,并将预测后的视频或图像在下方显示。
* 训练目前考虑到GPU等资源限制未完整开放。
* 信息显示在界面右下角显示类别flvgxmbwobject的识别目标个数。
2主要功能设计
设计了图片和视频两种数据输入的多目标识别功能。针对图片数据,调用模型进行单张图片预测,随后在前端可视化输出多目标识别结果;针对视频流动态图像数据,首先对视频流数据进行分帧采样,获取采样图片,再针对采样图片进行多目标识别,将采样识别结果进行视频合成,然后在前段可视化输出视频流数据识别结果。为求视频流数据处理的高效性,设计了采样-识别-展示的多线性处理方式,可加快视频流数据处理。
* 侧扫声呐图像多目标识别功能
* 侧扫声呐视频多目标识别功能
## 5. 软件使用结果
Faster-RCNN模型在四种目标物图片上的识别验证结果如下所示
![D:\pic\脉冲](https://osredm.com/repo/PulseFocusPlatform/PulseFocusPlatform/raw/branch/master/pic1/3.png)
YOLOV3模型在四种目标物图片上的识别验证结果如下所示
![D:\pic\脉冲](https://osredm.com/repo/PulseFocusPlatform/PulseFocusPlatform/raw/branch/master/pic1/4.png)
PP-YOLO-BOT模型在四种目标物图片上的识别验证结果如下所示
![D:\pic\脉冲](https://osredm.com/repo/PulseFocusPlatform/PulseFocusPlatform/raw/branch/master/pic1/5.png)
调用PP-YOLO-BOT模型对视频数据进行识别验证结果如下截图所示
![D:\pic\脉冲](https://osredm.com/repo/PulseFocusPlatform/PulseFocusPlatform/raw/branch/master/pic1/6.png)
## 6. 其他说明
* 使用GPU版本
参考百度飞桨paddle官方网站安装 参考百度飞桨paddle官方网站安装
[安装链接](https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/windows-pip.html) [安装链接](https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/windows-pip.html)
* 模型文件全部更新在inference_model中pic为测试图片 2. 模型文件全部更新在inference_model中pic为测试图片

View File

@ -1,11 +1,9 @@
# -*- coding: utf-8 -*-
# Form implementation generated from reading ui file 'SSS_win.ui' # Form implementation generated from reading ui file 'SSS_win.ui'
# #
# Created by: PyQt5 UI code generator 5.15.4 # Created by: PyQt5 UI code generator 5.12.3
# #
# WARNING: Any manual changes made to this file will be lost when pyuic5 is # WARNING! All changes made in this file will be lost!
# run again. Do not edit this file unless you know what you are doing.
from PyQt5 import QtCore, QtGui, QtWidgets from PyQt5 import QtCore, QtGui, QtWidgets
@ -92,6 +90,9 @@ class Ui_MainWindow(object):
self.pushButton_rjwendang = QtWidgets.QPushButton(self.centralwidget) self.pushButton_rjwendang = QtWidgets.QPushButton(self.centralwidget)
self.pushButton_rjwendang.setObjectName("pushButton_rjwendang") self.pushButton_rjwendang.setObjectName("pushButton_rjwendang")
self.verticalLayout_3.addWidget(self.pushButton_rjwendang) self.verticalLayout_3.addWidget(self.pushButton_rjwendang)
self.pushButton = QtWidgets.QPushButton(self.centralwidget)
self.pushButton.setObjectName("pushButton")
self.verticalLayout_3.addWidget(self.pushButton)
self.pushButton_tuichu = QtWidgets.QPushButton(self.centralwidget) self.pushButton_tuichu = QtWidgets.QPushButton(self.centralwidget)
self.pushButton_tuichu.setObjectName("pushButton_tuichu") self.pushButton_tuichu.setObjectName("pushButton_tuichu")
self.verticalLayout_3.addWidget(self.pushButton_tuichu) self.verticalLayout_3.addWidget(self.pushButton_tuichu)
@ -147,13 +148,13 @@ class Ui_MainWindow(object):
self.xunlian.clicked.connect(MainWindow.press_xunlian) self.xunlian.clicked.connect(MainWindow.press_xunlian)
self.pushButton_tuichu.clicked.connect(MainWindow.exit) self.pushButton_tuichu.clicked.connect(MainWindow.exit)
self.shipindaoru.clicked.connect(MainWindow.press_movie) self.shipindaoru.clicked.connect(MainWindow.press_movie)
self.comboBox_sbsuanfa.activated.connect(MainWindow.moxingxuanze) self.comboBox_sbsuanfa.activated['QString'].connect(MainWindow.moxingxuanze)
self.comboBox_GPU.activated.connect(MainWindow.gpu_use) self.comboBox_GPU.activated['QString'].connect(MainWindow.gpu_use)
QtCore.QMetaObject.connectSlotsByName(MainWindow) QtCore.QMetaObject.connectSlotsByName(MainWindow)
def retranslateUi(self, MainWindow): def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate _translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "脉冲聚焦")) MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
self.tupiandiaoru.setText(_translate("MainWindow", "静态图像导入")) self.tupiandiaoru.setText(_translate("MainWindow", "静态图像导入"))
self.shipindaoru.setText(_translate("MainWindow", "动态图像导入")) self.shipindaoru.setText(_translate("MainWindow", "动态图像导入"))
self.pushButton_xxdaochu.setText(_translate("MainWindow", "信息导出")) self.pushButton_xxdaochu.setText(_translate("MainWindow", "信息导出"))
@ -162,7 +163,7 @@ class Ui_MainWindow(object):
self.comboBox_yclfangfa.setItemText(0, _translate("MainWindow", "多尺度融合")) self.comboBox_yclfangfa.setItemText(0, _translate("MainWindow", "多尺度融合"))
self.comboBox_yclfangfa.setItemText(1, _translate("MainWindow", "图像增广")) self.comboBox_yclfangfa.setItemText(1, _translate("MainWindow", "图像增广"))
self.comboBox_yclfangfa.setItemText(2, _translate("MainWindow", "图像重塑")) self.comboBox_yclfangfa.setItemText(2, _translate("MainWindow", "图像重塑"))
self.label_3.setText(_translate("MainWindow", "聚焦算法")) self.label_3.setText(_translate("MainWindow", "识别算法"))
self.comboBox_sbsuanfa.setCurrentText(_translate("MainWindow", "PPYOLO-BOT")) self.comboBox_sbsuanfa.setCurrentText(_translate("MainWindow", "PPYOLO-BOT"))
self.comboBox_sbsuanfa.setItemText(0, _translate("MainWindow", "PPYOLO-BOT")) self.comboBox_sbsuanfa.setItemText(0, _translate("MainWindow", "PPYOLO-BOT"))
self.comboBox_sbsuanfa.setItemText(1, _translate("MainWindow", "YOLOV3")) self.comboBox_sbsuanfa.setItemText(1, _translate("MainWindow", "YOLOV3"))
@ -172,10 +173,11 @@ class Ui_MainWindow(object):
self.comboBox_GPU.setItemText(0, _translate("MainWindow", "YES")) self.comboBox_GPU.setItemText(0, _translate("MainWindow", "YES"))
self.comboBox_GPU.setItemText(1, _translate("MainWindow", "NO")) self.comboBox_GPU.setItemText(1, _translate("MainWindow", "NO"))
self.label.setText(_translate("MainWindow", "特征选择")) self.label.setText(_translate("MainWindow", "特征选择"))
self.shibie.setText(_translate("MainWindow", "聚焦")) self.shibie.setText(_translate("MainWindow", "识别"))
self.xunlian.setText(_translate("MainWindow", "训练")) self.xunlian.setText(_translate("MainWindow", "训练"))
self.pushButton_jswendang.setText(_translate("MainWindow", "技术文档")) self.pushButton_jswendang.setText(_translate("MainWindow", "技术文档"))
self.pushButton_rjwendang.setText(_translate("MainWindow", "软件说明文档")) self.pushButton_rjwendang.setText(_translate("MainWindow", "软件说明文档"))
self.pushButton.setText(_translate("MainWindow", "关于我们"))
self.pushButton_tuichu.setText(_translate("MainWindow", "退出")) self.pushButton_tuichu.setText(_translate("MainWindow", "退出"))
self.label_mb1.setText(_translate("MainWindow", "flv")) self.label_mb1.setText(_translate("MainWindow", "flv"))
self.flv.setText(_translate("MainWindow", "0")) self.flv.setText(_translate("MainWindow", "0"))

View File

@ -189,6 +189,13 @@
</property> </property>
</widget> </widget>
</item> </item>
<item>
<widget class="QPushButton" name="pushButton">
<property name="text">
<string>关于我们</string>
</property>
</widget>
</item>
<item> <item>
<widget class="QPushButton" name="pushButton_tuichu"> <widget class="QPushButton" name="pushButton_tuichu">
<property name="text"> <property name="text">

Binary file not shown.

18
main.py
View File

@ -1,6 +1,5 @@
import random import random
import string import string
# from typing_extensions import ParamSpecArgs
from PyQt5.QtWidgets import * from PyQt5.QtWidgets import *
from SSS_win import Ui_MainWindow from SSS_win import Ui_MainWindow
from PyQt5 import QtCore, QtGui, QtWidgets from PyQt5 import QtCore, QtGui, QtWidgets
@ -38,7 +37,7 @@ class mywindow(QtWidgets.QMainWindow, Ui_MainWindow):
self.stopEvent = threading.Event() self.stopEvent = threading.Event()
self.stopEvent.clear() self.stopEvent.clear()
# self.setWindowFlags(Qt.FramelessWindowHint)
def exit(self): def exit(self):
self.close() self.close()
@ -68,7 +67,6 @@ class mywindow(QtWidgets.QMainWindow, Ui_MainWindow):
print('cd {}'.format(self.path1)) print('cd {}'.format(self.path1))
print( print(
'python deploy/python/infer.py --model_dir={} --video_file={} --use_gpu=True'.format(self.model_path, self.Video_fname)) 'python deploy/python/infer.py --model_dir={} --video_file={} --use_gpu=True'.format(self.model_path, self.Video_fname))
# 调用GPU
os.system('cd {}'.format(self.path1)) os.system('cd {}'.format(self.path1))
os.system( os.system(
'python deploy/python/infer.py --model_dir={} --image_dir={} --output_dir=./video_output/{} --threshold=0.3 --use_gpu=True'.format( 'python deploy/python/infer.py --model_dir={} --image_dir={} --output_dir=./video_output/{} --threshold=0.3 --use_gpu=True'.format(
@ -126,20 +124,6 @@ class mywindow(QtWidgets.QMainWindow, Ui_MainWindow):
except Exception as e: except Exception as e:
pass pass
# files=os.listdir(self.video_image_path)
# print(type(files))
# files=self.Sorted(files)
# print(files)
# for xmlFile in files:
# print(xmlFile)
# self.label_movie.setPixmap(QtGui.QPixmap(
# self.video_image_path+'/'+xmlFile))
# time.sleep(0.5)
def Sorted(self,files): def Sorted(self,files):
files=[int(i.split('.')[0]) for i in files] files=[int(i.split('.')[0]) for i in files]
files.sort() files.sort()

Binary file not shown.

Before

Width:  |  Height:  |  Size: 26 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 19 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 360 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 350 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 372 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 942 KiB