s426206/WTV2D
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added video list page

Browse Source
This commit is contained in:
Jan Nowak 2022-10-30 00:26:33 +02:00
parent f82275df71
commit 3d4cf9daa8
7 changed files with 112 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.idea/vcs.xml
View File

@ -2,5 +2,6 @@
<project version="4"> <project version="4">
<component name="VcsDirectoryMappings"> <component name="VcsDirectoryMappings">
<mapping directory="$PROJECT_DIR$" vcs="Git" /> <mapping directory="$PROJECT_DIR$" vcs="Git" />
<mapping directory="$PROJECT_DIR$/TV2D" vcs="Git" />
</component> </component>
</project> </project>

0
README.md Normal file
View File

6
core/models.py
View File

@ -23,9 +23,15 @@ class VideoFile(models.Model):
video_width = models.IntegerField(default=0, blank=True) video_width = models.IntegerField(default=0, blank=True)
video_height = models.IntegerField(default=0, blank=True) video_height = models.IntegerField(default=0, blank=True)
status = models.IntegerField(default=VideoFileStatus.NEW.value) status = models.IntegerField(default=VideoFileStatus.NEW.value)
output_file = models.FileField(upload_to="", max_length=1024, blank=True)
csv_file = models.FileField(upload_to="", max_length=1024, blank=True)
@property @property
def url(self): def url(self):
return self.file.url return self.file.url
class WTV2D_data(models.Model):
processing = models.BooleanField(default=False, blank=False)

6
core/templates/core/home.html
View File

@ -12,6 +12,10 @@
<body> <body>
<div> <div>
<div class="d-flex flex-column min-vh-100 justify-content-center align-items-center"> <div class="d-flex flex-column min-vh-100 justify-content-center align-items-center">
{% if processing %}
<h1> Some video is still being processed</h1>
<a class="btn btn-primary" href="{% url 'videolist' %}">Video list</a>
{% else %}
<h1> Submit your video</h1> <h1> Submit your video</h1>
<form id="form" method="POST" class="mt-4 text-center" enctype="multipart/form-data"> <form id="form" method="POST" class="mt-4 text-center" enctype="multipart/form-data">
{% csrf_token %} {% csrf_token %}
@ -25,6 +29,8 @@
{{ form.file }} {{ form.file }}
<input type="submit" class="btn btn-primary mt-2" value="Upload"> <input type="submit" class="btn btn-primary mt-2" value="Upload">
</form> </form>
<a class="btn btn-primary mt-5" href="{% url 'videolist' %}">Video list</a>
{% endif %}
</div> </div>
</div> </div>
</body> </body>

53
core/templates/core/video_list.html Normal file
View File

@ -0,0 +1,53 @@
{% load static %}
<!DOCTYPE html>
<html>
<head>
<link href="https://cdn.jsdelivr.net/npm/bootstrap@5.0.2/dist/css/bootstrap.min.css" rel="stylesheet"
integrity="sha384-EVSTQN3/azprG1Anm3QDgpJLIm9Nao0Yz1ztcQTwFspd3yD65VohhpuuCOmLASjC" crossorigin="anonymous">
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
<script type="text/javascript" src="/static/js/jquery-3.6.0.min.js"></script>
<link rel="stylesheet" href="{% static 'css/main.css' %}">
<title>Web-TV2D - Video list</title>
</head>
<body>
<div class="container my-5">
<a class="btn btn-primary" href="{% url 'home' %}">Upload video</a>
<a class="btn btn-primary ms-5" href="{% url 'videolist' %}">Refresh</a>
<div class="d-flex flex-column justify-content-center align-items-center">
<table class="table">
<thead>
<tr>
<th scope="col">#</th>
<th scope="col">Video original file</th>
<th scope="col">Video status</th>
<th scope="col">Video processed file</th>
<th scope="col">Video CSV file</th>
</tr>
</thead>
<tbody>
{% for video in videos %}
<tr>
<th scope="row"> {{ video.pk }}</th>
<td><a href="{{ video.file.url }}">{{ video.file }}</a></td>
{% if video.status == enum_video_file_status.NEW.value %}
<td>NEW</td>
{% elif video.status == enum_video_file_status.PROCESSING.value %}
<td>PROCESSING</td>
{% elif video.status == enum_video_file_status.READY.value %}
<td>READY</td>
{% endif %}
{% if video.output_file and video.csv_file %}
<td><a href="{{ video.output_file.url }}">{{ video.output_file }}</a></td>
<td><a href="{{ video.csv_file.url }}">{{ video.csv_file }}</a></td>
{% else %}
<td></td>
<td></td>
{% endif %}
</tr>
{% endfor %}
</tbody>
</table>
</div>
</div>
</body>
</html>

5
core/urls.py
View File

@ -2,8 +2,7 @@ from django.contrib import admin
from django.urls import path from django.urls import path
from . import views from . import views
app_name = "menu"
urlpatterns = [ urlpatterns = [
path('' , views.home, name="home"), path('/', views.home, name="home"),
path('videolist/', views.video_list_view, name="videolist"),
] ]

45
core/views.py
View File

@ -3,27 +3,70 @@ from django.http import HttpResponseRedirect
from django.shortcuts import render from django.shortcuts import render
from django.contrib import messages from django.contrib import messages
from core.forms import VideoUploadForm from core.forms import VideoUploadForm
from core.models import VideoFile from core.models import VideoFile, WTV2D_data
from TV2D import TV2D
# Create your views here. # Create your views here.
def home(request): def home(request):
wtv2d_data = WTV2D_data.objects.first()
# For control processing videos some workers should be used like Celery
# if wtv2d_data.processing:
# context = {'processing': True}
# return render(request, 'core/home.html', context)
form = VideoUploadForm() form = VideoUploadForm()
if request.method == "POST": if request.method == "POST":
form = VideoUploadForm(request.POST, request.FILES) form = VideoUploadForm(request.POST, request.FILES)
if form.is_valid(): if form.is_valid():
wtv2d_data.processing = True
wtv2d_data.save()
video_file = VideoFile(file=request.FILES['file']) video_file = VideoFile(file=request.FILES['file'])
video_file.status = VideoFile.VideoFileStatus.NEW.value
video_file.save() video_file.save()
video = cv2.VideoCapture(video_file.file.path) video = cv2.VideoCapture(video_file.file.path)
video_file.fps = video.get(cv2.CAP_PROP_FPS) video_file.fps = video.get(cv2.CAP_PROP_FPS)
video_file.video_width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH)) video_file.video_width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
video_file.video_height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT)) video_file.video_height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
video_file.status = VideoFile.VideoFileStatus.PROCESSING.value
video_file.save() video_file.save()
# object_detection_model_path = "COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yaml"
object_detection_model_path = "COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml"
# Same as model by for now for config. If you use model from local sotrage you can still use config
## from Detectron2 Model Zoo and Baselines.
# object_detection_config_path = "COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yaml"
object_detection_config_path = "COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml"
homography_keypoint_path = "TV2D/models/FPN_efficientnetb3_0.0001_4.h5"
homography_deephomo_path = "TV2D/models/HomographyModel_0.0001_4.h5"
deep_sort_model = "TV2D/models/market_bot_R50.pth"
deep_sort_model_config = "TV2D/deep_sort_pytorch/thirdparty/fast-reid/configs/Market1501/bagtricks_R50.yml"
tv2d = TV2D.TV2D(object_detection_model_path, object_detection_config_path=object_detection_config_path,
homography_on=True, team_detection_on=True,
tracker_on=True, no_gui=True,
homography_pretreined=False, homography_deephomo_path=homography_deephomo_path,
homography_keypoint_path=homography_keypoint_path,
deep_sort_model_path=deep_sort_model, deep_sort_model_config=deep_sort_model_config)
output_video_name = ".".join(video_file.file.name.split(".")[:-1]) + "_" + str(video_file.pk) + ".mkv"
export_data_file_path = ".".join(video_file.file.name.split(".")[:-1]) + "_" + str(video_file.pk) + ".csv"
video_file.output_file = output_video_name
video_file.csv_file = export_data_file_path
video_file.save()
tv2d(TV2D.TV2D.RunOn.VIDEO, video_file.file.path, export_output_path=f"media/{output_video_name}",
export_data_file_path=f"media/{export_data_file_path}")
video_file.status = VideoFile.VideoFileStatus.READY.value
video_file.save()
else: else:
messages.error(request, "Something went wrong") messages.error(request, "Something went wrong")
wtv2d_data.processing = False
wtv2d_data.save()
context = {'form': form} context = {'form': form}
return render(request, 'core/home.html', context) return render(request, 'core/home.html', context)
def video_list_view(request):
videos = VideoFile.objects.all().order_by("-pk")
context = {'videos': videos, 'enum_video_file_status': VideoFile.VideoFileStatus}
return render(request, 'core/video_list.html', context)