93 lines
12 KiB
Plaintext
93 lines
12 KiB
Plaintext
|
o
|
|||
|
|
¦<<i_> ã @ sb d Z ddlmZmZmZ ddlmZ ddlmZ ddl m
|
|||
|
|
Z
|
|||
|
|
m Z ddl m
|
|||
|
|
Z
|
|||
|
|
ddlmZmZ ddlZddlZddlZdd lmZ dd
|
|||
|
|
lmZ ddlZed d d
|
|||
|
|
�Zejedgddgdgd� ddddddddœaddddddddœaG dd„ de
|
|||
|
|
ƒZ G dd„ de
|
|||
|
|
ƒZ G dd„ de
|
|||
|
|
ƒZ ej de
|
|||
|
|
d �d d „ ƒZ e d ¡d d!„ ƒZ!ej d"e d#�d$d%„ ƒZ"e #d&¡d'e d(efd)d*„ƒZ$e #d+¡d,d-„ ƒZ%e d.¡d/d0„ ƒZ&e #d1¡d'e d(efd2d3„ƒZ'e d4¡d5d6„ ƒZ(d'e fd7d8„Z)d9e*fd:d;„Z+d9e*fd<d=„Z,d'e fd>d?„Z-e.d@k�r/e/dAƒ e/dBƒ e/dAƒ e/dCƒ e/dDƒ e/dEƒ e/dFƒ e/dGƒ e/dHƒ ej0edIdJdKdL� dS dS )Mu}
|
|||
|
|
API Server for Land Classification Model Training
|
|||
|
|
Cho phép ch�n dữ liệu và cấu hình training qua giao diện web
|
|||
|
|
é )ÚFastAPIÚBackgroundTasksÚ
|
|||
|
|
HTTPException)ÚCORSMiddleware)Ú StaticFiles)Ú HTMLResponseÚ FileResponse)Ú BaseModel)ÚOptionalÚListN©Údatetime)ÚPathz Land Classification Training APIz1.0.0)ÚtitleÚversionÚ*T)Z
|
|||
|
|
allow_originsZallow_credentialsZ
|
|||
|
|
allow_methodsZ
|
|||
|
|
allow_headersFÚ )Ú is_trainingÚprogressÚerrorÚresultÚ
|
|||
|
|
start_timeÚend_timeÚcancel_requested©Ú
|
|||
|
|
is_predictingr r r Ú output_filer r c @ sº e Zd ZU dZdZeed< dZeed< dZeed< dZ eed < d
|
|||
|
|
Z
|
|||
|
|
e ed < d Z e ed
|
|||
|
|
< dZ
|
|||
|
|
eed< dZeed< dZeed< dZeed< dZeed< dZeed< dZeed< dZe ed < d S ) ÚTrainingConfigu Cấu hình trainingçffffffZ@Úmin_lonçš™™™™™"@Úmin_latçÍÌÌÌÌŒZ@Úmax_lonçš™™™™™#@Úmax_latú
|
|||
|
|
2023-03-01Ú
|
|||
|
|
start_dateú
|
|||
|
|
2023-05-31Úend_dateé Ú
|
|||
|
|
max_scenesé Ú cloud_coveré Ú
|
|||
|
|
resolutionéd Ú n_estimatorsÚ max_depthgš™™™™™¹?Ú
|
|||
|
|
learning_rateTÚuse_gpuz1train/ST_training data_updated_1130points_new.shpÚtraining_shapefileN)Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ú__doc__r ÚfloatÚ__annotations__r! r# r% r' Ústrr) r+ Úintr- r/ r1 r2 r3 r4 Úboolr5 © r? r? ú1/home/x79/CSIROBoeingPhase5-Vietnam/api_server.pyr 5 s
|
|||
|
|
r c @ s† e Zd ZU dZeed< dZeed< dZeed< dZ eed< d Z
|
|||
|
|
eed
|
|||
|
|
< d Z eed < d
|
|||
|
|
Z eed< dZ
|
|||
|
|
eed< dZeed< dZeed< dS )ÚPredictionConfigu Cấu hình dá»± Ä‘oánÚmodel_filenamer r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ N)r6 r7 r8 r9 r< r; r r: r! r# r% r' r) r+ r= r- r/ r? r? r? r@ rA P s
|
|||
|
|
rA c @ sR e Zd ZU dZeed< eed< ee ed< ee ed< ee ed< ee ed< dS ) ÚTrainingStatusu Trạng thái trainingr r r r r r N) r6 r7 r8 r9 r> r; r< r
|
|||
|
|
Údictr? r? r? r@ rC e s
|
|||
|
|
rC ú/)Zresponse_classc à s( �t tƒjd } | ¡ rt| ƒS tdƒS )u Serve giao diện webztraining_interface.htmluV
|
|||
|
|
<html>
|
|||
|
|
<head><title>Training Interface</title></head>
|
|||
|
|
<body>
|
|||
|
|
<h1>Land Classification Training API</h1>
|
|||
|
|
<p>API Documentation: <a href="/docs">/docs</a></p>
|
|||
|
|
<p>Training Interface: Tạo file training_interface.html</p>
|
|||
|
|
</body>
|
|||
|
|
</html>
|
|||
|
|
)r Ú__file__ÚparentÚexistsr r )Z html_filer? r? r@ Úrooto s
|
|||
|
|
€rI z/api/config/presetsc à s^ �ddddddddd d
|
|||
|
|
d d œ d
|
|||
|
|
œdddddddd
|
|||
|
|
d
|
|||
|
|
dd œ d
|
|||
|
|
œddddddddddd œ d
|
|||
|
|
œgiS )u# Lấy các preset cấu hình sẵnZpresetsu% PC - Nhá»� (3 tháng, 20m, 12 scenes)r r r" r$ r& r( r* r, r. ) r r! r# r% r' r) r+ r- r/ )ÚnameÚconfigu/ Server - Trung bình (6 tháng, 10m, 30 scenes)g `Z@gffffff"@gš™™™™™Z@g $@z
|
|||
|
|
2023-01-01z
|
|||
|
|
2023-06-30é
|
|||
|
|
u% Full - Lớn (1 năm, 10m, 60 scenes)z
|
|||
|
|
2022-09-01z
|
|||
|
|
2023-10-01é< é2 r? r? r? r? r@ Ú get_presets‚ s, €ýþ ýþ ýþïÿrO z/api/training/status)Zresponse_modelc à ó �t S )u! Kiểm tra trạng thái training©Útraining_statusr? r? r? r@ Úget_training_status£ ó €rS z/api/training/startrK Úbackground_tasksc à sF �t d r tddd�‚ddddt ¡ ¡ ddœa | t| ¡ d t d
|
|||
|
|
œS ) u. Bắt đầu training vá»›i config đã chá»�nr é� u' Training Ä‘ang chạy, vui lòng đợi©Ú status_codeZdetailTu Ä�ang khởi tạo...N)r r r r r r u Training đã bắt đầu©ÚmessageÚstatus)rR r r
|
|||
|
|
ÚnowÚ isoformatÚadd_taskÚ run_training©rK rU r? r? r@ Ústart_training© s €
|
|||
|
|
ú
|
|||
|
|
|
|||
|
|
ra z/api/training/stopc à s* �t d s ddiS dt d< dt d< ddiS ) u$ Dừng training (nếu Ä‘ang chạy)r rZ u% Không có training nà o Ä‘ang chạyTr u Ä�ang hủy training...r u Ä�ang dừng training...rQ r? r? r? r@ Ú
|
|||
|
|
stop_trainingÁ s €rb z/api/models/listc à sÊ �t dƒ} | ¡ s
|
|||
|
|
dg iS g }| d¡D ]C}| d¡}i }| ¡ r:t|ƒ�
|
|||
|
|
}t |¡}W d ƒ n1 s5w Y | |jt
|
|||
|
|
| ¡ j ¡
|
|||
|
|
¡ t| ¡ jd d dƒ|dœ¡ q|jd d
|
|||
|
|
„ d d � d|iS )
|
|||
|
|
u Liệt kê các model đã trainÚ model_trainÚmodelsz*.joblibz.jsonNi é )ÚfilenameÚcreatedZsize_mbÚinfoc S s | d S )Nrg r? )Úxr? r? r@ Ú<lambda>ç ó z list_models.<locals>.<lambda>T)ÚkeyÚreverse)r rH ÚglobÚ with_suffixÚopenÚjsonÚloadÚappendrJ r
|
|||
|
|
Ú
|
|||
|
|
fromtimestampÚstatÚst_mtimer] ÚroundÚst_sizeÚsort)Z model_dirrd Z
|
|||
|
|
model_fileZ info_filerh Úfr? r? r@ Ú list_modelsÐ s( €
|
|||
|
|
|
|||
|
|
ÿ
|
|||
|
|
ür{ z/api/prediction/startc à sH �t d r tddd�‚dddddt ¡ ¡ ddœa | t| ¡ d t d
|
|||
|
|
œS ) u Bắt đầu dá»± Ä‘oánr rV u( Ä�ang có dá»± Ä‘oán khác Ä‘ang chạyrW Tu Ä�ang khởi Ä‘á»™ng...Nr u Ä�ã bắt đầu dá»± Ä‘oánrY )Úprediction_statusr r
|
|||
|
|
r\ r] r^ Úrun_predictionr` r? r? r@ Ústart_predictionë s €
|
|||
|
|
ù
|
|||
|
|
r~ z/api/prediction/statusc à rP )u$ Kiểm tra trạng thái dá»± Ä‘oán)r| r? r? r? r@ Úget_prediction_status rT r c à sF �zidt d< dt d< ddlm} dt d< dd „ }|| j| j| j| jg| j› d
|
|||
|
|
| j› �| j | j
|
|||
|
|
| j | j | j
|
|||
|
|
| j| j| jd d „ |d
|
|||
|
|
� }t dd¡rTdt d< dt d< dt d< n dt d< dt d< |t d< t ¡ ¡ t d< W dS ty¢ } z,dt d< t|ƒt d< dt|ƒ› �t d< t ¡ ¡ t d< ddl}t| ¡ ƒ W Y d}~dS d}~ww )u Chạy training processFr õ Ä�ang import thÆ° viện...r r )Ú train_modelu$ Ä�ang load dữ liệu Sentinel-2...c S s t dd¡S )Nr F)rR Úgetr? r? r? r@ Ú
|
|||
|
|
should_cancel s z#run_training.<locals>.should_cancelrE c S s t | ƒS )N)Úupdate_progress)Úmsgr? r? r@ rj ' rk z run_training.<locals>.<lambda>) ÚbboxÚ
|
|||
|
|
time_ranger+ r- r/ r5 r1 r2 r3 r4 Zstatus_callbackZ cancel_checkr u Ä�ã hủy trainingzTraining cancelled by userr õ
|
|||
|
|
Hoà n thà nh!r r õ Lá»—i: N)rR Z train_moduler� r r! r# r% r' r) r+ r- r/ r5 r1 r2 r3 r4 r‚ r
|
|||
|
|
r\ r] Ú Exceptionr< Ú tracebackÚprintÚ
|
|||
|
|
format_exc)rK r� rƒ r Úer‹ r? r? r@ r_
|
|||
|
|
sL € ô
|