1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24o
�ߜgP � @ sN d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ G dd� d�Z dS ) � )�wraps)�futures)�tqdm)�Path)�Imagec @ s: e Zd Zdd� Zdd� Zd
dd�Zdd d
�Zdd d �ZdS )�
ProcessingAPIc C s
|| _ d S �N)�data)�selfr � r �@/mnt/f/workspace/GIStudio/SVIkit/SVIkit/src/svikit/processing.py�__init__ s
zProcessingAPI.__init__c sz g }t �� �&� � ��fdd�| jjD �}tt �|��D ] }|�|�� � qW d � n1 s/w Y || jj�j< | S )Nc s$ g | ]}� j �|d fi ����qS )�path)�submit)�.0�img��executor�func�kwargsr r �
<listcomp> s �z'ProcessingAPI.apply.<locals>.<listcomp>) r �ThreadPoolExecutorr r � as_completed�append�result� processed�__name__)r
r r �resultsZ futures_list�futurer r r �apply s
�� �zProcessingAPI.applyNc C s dd� }| � |p |�S )Nc S s d S r r )r r r r �default_segment s z3ProcessingAPI.segmentation.<locals>.default_segment)r )r
�modelr r r r � segmentation s zProcessingAPI.segmentation�� r$ c C s& | j | j|d�}td| jj� �� |S )u$ 生成缩略图并返回处理结果)� output_sizeu 处理结果: )r �_create_thumbnail�printr r )r
r% r r r r �create_thumbnail s z ProcessingAPI.create_thumbnailc
C s zJt |��� rt |��� jdkrtd|� �� W dS t�|�}|�|� t |�jd dt |�j � � }|jj
dd� |� |� t |�t |�|j
d�W S tyg } ztd |� d
|� �� W Y d}~dS d}~w ty� } ztd |� d
|� �� W Y d}~dS d}~ww ) u 生成缩略图的具体实现r u 文件不存在或为空: NZthumbsZthumb_T)�exist_ok)Z
original_pathZthumbnail_pathZprocessed_sizeu 处理图像时出错: u - 错误信息: u# 处理图像时发生未知错误: )r �exists�stat�st_sizer' r �open� thumbnail�parent�name�mkdir�save�str�size�OSError� Exception)r
Zimg_pathr% r Z
thumb_path�er r r r& % s,
����z ProcessingAPI._create_thumbnailr )r# )r �
__module__� __qualname__r
r r" r( r& r r r r r s
r N)
� functoolsr �
concurrentr r �pathlibr �PILr r r r r r �<module> s