forked from prehistoric-systems/comixify
1e5252b8f0
* Add ComixGAN #1 * Add minor fixes * Reduce batch size * Fix concatenate bug * Fix GPU memory not released problem #2 * Build client * Fix style_transfer_mode bug * Improve timings * Add minor fix with GPU name * Fix tf session problem * Compile frontend * Change parameters * Fix occasional yt Error
32 lines
1.7 KiB
Python
32 lines
1.7 KiB
Python
from django.conf import settings
|
|
from rest_framework import serializers
|
|
|
|
from .exceptions import FileExtensionError, TooLargeFile
|
|
|
|
|
|
class VideoSerializer(serializers.Serializer):
|
|
file = serializers.FileField()
|
|
frames_mode = serializers.IntegerField(min_value=0, max_value=1, default=settings.DEFAULT_FRAMES_SAMPLING_MODE)
|
|
rl_mode = serializers.IntegerField(min_value=0, max_value=1, default=settings.DEFAULT_RL_MODE)
|
|
image_assessment_mode = serializers.IntegerField(min_value=0, max_value=1,
|
|
default=settings.DEFAULT_IMAGE_ASSESSMENT_MODE)
|
|
style_transfer_mode = serializers.IntegerField(min_value=0, max_value=2,
|
|
default=settings.DEFAULT_STYLE_TRANSFER_MODE)
|
|
|
|
def validate(self, attrs):
|
|
file = attrs.get("file")
|
|
if file.name.split(".")[-1] not in settings.PERMITTED_VIDEO_EXTENSIONS:
|
|
raise FileExtensionError
|
|
if file.size > settings.MAX_FILE_SIZE:
|
|
raise TooLargeFile
|
|
return attrs
|
|
|
|
|
|
class YouTubeDownloadSerializer(serializers.Serializer):
|
|
url = serializers.URLField()
|
|
frames_mode = serializers.IntegerField(min_value=0, max_value=1, default=settings.DEFAULT_FRAMES_SAMPLING_MODE)
|
|
rl_mode = serializers.IntegerField(min_value=0, max_value=1, default=settings.DEFAULT_RL_MODE)
|
|
image_assessment_mode = serializers.IntegerField(min_value=0, max_value=1,
|
|
default=settings.DEFAULT_IMAGE_ASSESSMENT_MODE)
|
|
style_transfer_mode = serializers.IntegerField(min_value=0, max_value=2,
|
|
default=settings.DEFAULT_STYLE_TRANSFER_MODE)
|