Video 和 Mark逻辑部分
This commit is contained in:
parent
f58a6a1866
commit
512517f62f
109
pdm.lock
generated
109
pdm.lock
generated
@ -5,11 +5,22 @@
|
||||
groups = ["default"]
|
||||
strategy = ["inherit_metadata"]
|
||||
lock_version = "4.5.0"
|
||||
content_hash = "sha256:22e7fb6be688755243be4decdc34c7a192d0929d9c4e6e62980ee8a44a8f2bc1"
|
||||
content_hash = "sha256:f6e9aeeb24f366560659a66700c5d5a228c427ef73f5f849ef55242aad81d427"
|
||||
|
||||
[[metadata.targets]]
|
||||
requires_python = "==3.12.*"
|
||||
|
||||
[[package]]
|
||||
name = "et-xmlfile"
|
||||
version = "1.1.0"
|
||||
requires_python = ">=3.6"
|
||||
summary = "An implementation of lxml.xmlfile for the standard library"
|
||||
groups = ["default"]
|
||||
files = [
|
||||
{file = "et_xmlfile-1.1.0-py3-none-any.whl", hash = "sha256:a2ba85d1d6a74ef63837eed693bcb89c3f752169b0e3e7ae5b16ca5e1b3deada"},
|
||||
{file = "et_xmlfile-1.1.0.tar.gz", hash = "sha256:8eb9e2bc2f8c97e37a2dc85a09ecdcdec9d8a396530a6d5a33b30b9a92da0c5c"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "numpy"
|
||||
version = "2.1.1"
|
||||
@ -58,6 +69,45 @@ files = [
|
||||
{file = "opencv_python-4.10.0.84-cp37-abi3-win_amd64.whl", hash = "sha256:32dbbd94c26f611dc5cc6979e6b7aa1f55a64d6b463cc1dcd3c95505a63e48fe"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "openpyxl"
|
||||
version = "3.1.5"
|
||||
requires_python = ">=3.8"
|
||||
summary = "A Python library to read/write Excel 2010 xlsx/xlsm files"
|
||||
groups = ["default"]
|
||||
dependencies = [
|
||||
"et-xmlfile",
|
||||
]
|
||||
files = [
|
||||
{file = "openpyxl-3.1.5-py2.py3-none-any.whl", hash = "sha256:5282c12b107bffeef825f4617dc029afaf41d0ea60823bbb665ef3079dc79de2"},
|
||||
{file = "openpyxl-3.1.5.tar.gz", hash = "sha256:cf0e3cf56142039133628b5acffe8ef0c12bc902d2aadd3e0fe5878dc08d1050"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pandas"
|
||||
version = "2.2.3"
|
||||
requires_python = ">=3.9"
|
||||
summary = "Powerful data structures for data analysis, time series, and statistics"
|
||||
groups = ["default"]
|
||||
dependencies = [
|
||||
"numpy>=1.22.4; python_version < \"3.11\"",
|
||||
"numpy>=1.23.2; python_version == \"3.11\"",
|
||||
"numpy>=1.26.0; python_version >= \"3.12\"",
|
||||
"python-dateutil>=2.8.2",
|
||||
"pytz>=2020.1",
|
||||
"tzdata>=2022.7",
|
||||
]
|
||||
files = [
|
||||
{file = "pandas-2.2.3-cp312-cp312-macosx_10_9_x86_64.whl", hash = "sha256:b1d432e8d08679a40e2a6d8b2f9770a5c21793a6f9f47fdd52c5ce1948a5a8a9"},
|
||||
{file = "pandas-2.2.3-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:a5a1595fe639f5988ba6a8e5bc9649af3baf26df3998a0abe56c02609392e0a4"},
|
||||
{file = "pandas-2.2.3-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:5de54125a92bb4d1c051c0659e6fcb75256bf799a732a87184e5ea503965bce3"},
|
||||
{file = "pandas-2.2.3-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:fffb8ae78d8af97f849404f21411c95062db1496aeb3e56f146f0355c9989319"},
|
||||
{file = "pandas-2.2.3-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:6dfcb5ee8d4d50c06a51c2fffa6cff6272098ad6540aed1a76d15fb9318194d8"},
|
||||
{file = "pandas-2.2.3-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:062309c1b9ea12a50e8ce661145c6aab431b1e99530d3cd60640e255778bd43a"},
|
||||
{file = "pandas-2.2.3-cp312-cp312-win_amd64.whl", hash = "sha256:59ef3764d0fe818125a5097d2ae867ca3fa64df032331b7e0917cf5d7bf66b13"},
|
||||
{file = "pandas-2.2.3.tar.gz", hash = "sha256:4f18ba62b61d7e192368b84517265a99b4d7ee8912f8708660fb4a366cc82667"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pyside6"
|
||||
version = "6.7.3"
|
||||
@ -109,6 +159,30 @@ files = [
|
||||
{file = "PySide6_Essentials-6.7.3-cp39-abi3-win_amd64.whl", hash = "sha256:f0950fcdcbcd4f2443336dc6a5fe692172adc225f876839583503ded0ab2f2a7"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "python-dateutil"
|
||||
version = "2.9.0.post0"
|
||||
requires_python = "!=3.0.*,!=3.1.*,!=3.2.*,>=2.7"
|
||||
summary = "Extensions to the standard Python datetime module"
|
||||
groups = ["default"]
|
||||
dependencies = [
|
||||
"six>=1.5",
|
||||
]
|
||||
files = [
|
||||
{file = "python-dateutil-2.9.0.post0.tar.gz", hash = "sha256:37dd54208da7e1cd875388217d5e00ebd4179249f90fb72437e91a35459a0ad3"},
|
||||
{file = "python_dateutil-2.9.0.post0-py2.py3-none-any.whl", hash = "sha256:a8b2bc7bffae282281c8140a97d3aa9c14da0b136dfe83f850eea9a5f7470427"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pytz"
|
||||
version = "2024.2"
|
||||
summary = "World timezone definitions, modern and historical"
|
||||
groups = ["default"]
|
||||
files = [
|
||||
{file = "pytz-2024.2-py2.py3-none-any.whl", hash = "sha256:31c7c1817eb7fae7ca4b8c7ee50c72f93aa2dd863de768e1ef4245d426aa0725"},
|
||||
{file = "pytz-2024.2.tar.gz", hash = "sha256:2aa355083c50a0f93fa581709deac0c9ad65cca8a9e9beac660adcbd493c798a"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "shiboken6"
|
||||
version = "6.7.3"
|
||||
@ -121,3 +195,36 @@ files = [
|
||||
{file = "shiboken6-6.7.3-cp39-abi3-manylinux_2_31_aarch64.whl", hash = "sha256:f0dd635178e64a45be2f84c9f33dd79ac30328da87f834f21a0baf69ae210e6e"},
|
||||
{file = "shiboken6-6.7.3-cp39-abi3-win_amd64.whl", hash = "sha256:5f29325dfa86fde0274240f1f38e421303749d3174ce3ada178715b5f4719db9"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "six"
|
||||
version = "1.16.0"
|
||||
requires_python = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*"
|
||||
summary = "Python 2 and 3 compatibility utilities"
|
||||
groups = ["default"]
|
||||
files = [
|
||||
{file = "six-1.16.0-py2.py3-none-any.whl", hash = "sha256:8abb2f1d86890a2dfb989f9a77cfcfd3e47c2a354b01111771326f8aa26e0254"},
|
||||
{file = "six-1.16.0.tar.gz", hash = "sha256:1e61c37477a1626458e36f7b1d82aa5c9b094fa4802892072e49de9c60c4c926"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tzdata"
|
||||
version = "2024.2"
|
||||
requires_python = ">=2"
|
||||
summary = "Provider of IANA time zone data"
|
||||
groups = ["default"]
|
||||
files = [
|
||||
{file = "tzdata-2024.2-py2.py3-none-any.whl", hash = "sha256:a48093786cdcde33cad18c2555e8532f34422074448fbc874186f0abd79565cd"},
|
||||
{file = "tzdata-2024.2.tar.gz", hash = "sha256:7d85cc416e9382e69095b7bdf4afd9e3880418a2413feec7069d533d6b4e31cc"},
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "xlrd"
|
||||
version = "2.0.1"
|
||||
requires_python = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*, !=3.5.*"
|
||||
summary = "Library for developers to extract data from Microsoft Excel (tm) .xls spreadsheet files"
|
||||
groups = ["default"]
|
||||
files = [
|
||||
{file = "xlrd-2.0.1-py2.py3-none-any.whl", hash = "sha256:6a33ee89877bd9abc1158129f6e94be74e2679636b8a205b43b85206c3f0bbdd"},
|
||||
{file = "xlrd-2.0.1.tar.gz", hash = "sha256:f72f148f54442c6b056bf931dbc34f986fd0c3b0b6b5a58d013c9aef274d0c88"},
|
||||
]
|
||||
|
||||
@ -9,6 +9,9 @@ dependencies = [
|
||||
"pyside6>=6.7.3",
|
||||
"opencv-python>=4.10.0.84",
|
||||
"numpy>=2.1.1",
|
||||
"pandas>=2.2.3",
|
||||
"openpyxl>=3.1.5",
|
||||
"xlrd>=2.0.1",
|
||||
]
|
||||
requires-python = "==3.12.*"
|
||||
readme = "README.md"
|
||||
@ -20,4 +23,4 @@ distribution = false
|
||||
|
||||
[tool.pdm.scripts]
|
||||
main = "python src/esd_bleeding_view_tool/__init__.py"
|
||||
nox = "python src/esd_bleeding_view_tool/non_ui.py"
|
||||
nox = "python src/esd_bleeding_view_tool/core.py"
|
||||
66
src/esd_bleeding_view_tool/core.py
Normal file
66
src/esd_bleeding_view_tool/core.py
Normal file
@ -0,0 +1,66 @@
|
||||
import os
|
||||
import cv2
|
||||
import pandas as pd
|
||||
import re
|
||||
|
||||
video_dir = "raw/video/王传林.mp4"
|
||||
mark_dir = "raw/mark/ESD-Bleeding-标注.xlsx"
|
||||
interval = 30
|
||||
|
||||
def convert_video(videoname):
|
||||
capture = cv2.VideoCapture(videoname)
|
||||
total_frames = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
|
||||
if capture.isOpened():
|
||||
frame_counter = 0
|
||||
if not os.path.exists("temp/"):
|
||||
os.makedirs("temp/")
|
||||
while True:
|
||||
ret, img = capture.read() # img 就是一帧图片
|
||||
if frame_counter % interval == 0:
|
||||
cv2.imwrite("temp/"+str(frame_counter)+".jpg", img)
|
||||
frame_counter += 1
|
||||
|
||||
if not ret:
|
||||
break # 当获取完最后一帧就结束
|
||||
else:
|
||||
print('视频打开失败！')
|
||||
|
||||
class Video:
|
||||
def __init__(self, video_dir):
|
||||
self.video_dir = video_dir
|
||||
capture = cv2.VideoCapture(video_dir)
|
||||
if capture.isOpened():
|
||||
self.total_frames = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
|
||||
self.video = capture
|
||||
else:
|
||||
self.video = None
|
||||
def read_frame(self, frame_num):
|
||||
cap = self.video
|
||||
cap.set(cv2.CAP_PROP_POS_FRAMES, frame_num)
|
||||
ret, img=cap.read()
|
||||
if ret:
|
||||
return img
|
||||
else:
|
||||
return None
|
||||
|
||||
class Mark:
|
||||
def __init__(self, mark_dir, sheet_name):
|
||||
self.df = pd.read_excel(mark_dir, sheet_name=sheet_name)
|
||||
col_name = self.df.columns.tolist()
|
||||
self.method = col_name[1:]
|
||||
self.frame = self.df.loc[:,col_name[0]]
|
||||
self.total_frame = len(self.frame)
|
||||
|
||||
def read(self, number):
|
||||
line = self.df.loc[number,:]
|
||||
rt = re.findall(r'\d+', line.iloc[0])
|
||||
if len(rt) > 0:
|
||||
return (int(rt[0]), line[self.method[0]:])
|
||||
else:
|
||||
return None
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
mark = Mark(mark_dir, "王传林")
|
||||
frame, method = mark.read(1)
|
||||
print(int(frame),method)
|
||||
@ -1,21 +0,0 @@
|
||||
import cv2
|
||||
|
||||
video_dir = "raw/video/王传林.mp4"
|
||||
mark_dir = "raw/mark/ESD-Bleeding-标注.xlsx"
|
||||
|
||||
def convert_video(videoname):
|
||||
capture = cv2.VideoCapture(videoname)
|
||||
if capture.isOpened():
|
||||
frame_counter = 0
|
||||
while True:
|
||||
ret, img = capture.read() # img 就是一帧图片
|
||||
if frame_counter % 30 == 0:
|
||||
cv2.imwrite("temp/"+str(frame_counter)+".jpg", img)
|
||||
frame_counter += 1
|
||||
|
||||
if not ret:break # 当获取完最后一帧就结束
|
||||
else:
|
||||
print('视频打开失败！')
|
||||
|
||||
if __name__ == "__main__":
|
||||
convert_video(video_dir)
|
||||
Loading…
x
Reference in New Issue
Block a user