K4qodeo3(gIv+#wQ(ESON4btL@-G>A$E8ims(X7|e$Vb#jB7Ee
z=jZO9PuSFq*IMS=M-?@a_8yUWbQ6D3#*1@EuoX^;0%#a9$UOnQbc`Y`ClKaMmr-h^
zhgQ`9r%Rg->>dzjFg0}rB^3=PrNn7|a6>iD8ZEN1x`Tx=mO$`zDP%CNyPvYN%jx<@RPDRW|__?Gk(qL^sCBVx)04t(_9P
zAoD}6*4{I%)l1JJ`KEfk^lm(|ASqc~++kOLvzT&Z;w?9Z^!(+U9awI6Pp3FqqAKpxT1?3yTLrgk;D#YUt(VqX19ff;1|K1&U0?=2q*nX&BYqs9Rz
z6QVG+;RBz*AK&rDTub#~ktR{20DU|!A$wO>k_~3Olf6PSD`8HEXtt$3Pt6EqTXWua)AI}Mm2s9GBS!2v#CDXFZPi74`Oz0C&
zWE;8CQmdRnH<&;JyT3y|enMZxnNcHzjU~i)loxdlfam5JeU6~XJp6UV6;BVq?G6Il
zQstTLmTHvwCF^GLr%6Y7$dxoR7{od!%VVCcb!fhr%QCaEKYMTI<7VM2&k)06KZ6*z
z+BZj<*g%F7H6*l!|12wzUM1QA5AOms)z4eV*%pQ#1b}jLiTbqy{oK9ou2Nal*)Mk3
z0WI$)LVMUm%vNrrsQIt&*}g1U19x>ob{RK`#;XBpK~Rn_xrpTvXB~jv$I1I|*UaH@
zm|GxwcISiX*aQDn`HzBE{Fmwd#VzV?o}dCp%h=8{vJTpP!MR)q?V|HpBRZC$NBt07
zk3)6wsFJGaj*h}m=^!o`K`h+`0g@h+U9B?{4@?s`aV4IL!C%TJ&YQm56MjN#>YNdGOTbQ6nk8$vKI#
zo)O)d_HmG~_CQ>!L%~MEG_HEuD!)o_GB1J}N=n8!a3K0`YJQN>AKYaNI5*DrqIc%pn
zKhSx1*0OL%fpOlE+~Fn4x+YNuL1zs=xnB2wlca>dN(qPN4b`
z+37?;PjnvPZchDW(98c(3jp5JV(J;4BRV}pAzzbxy=J?CG4VH@+Ht1Fx39(DE
zylckQ@6pjPLR|i5`YStJ`w~p)*k!lxj*#`;I>P#xAPf`4Pd6bqy_j1VJsXf-yqyyn
zppdqJ?q>?cP>Z(bi52-X9zHhHCX>JJchF~NSjvfgC1RGh;`em-;`l*kv*vW4kfCQ4
z{jHGo#4{uBp?ix1kv}=i3*vXV7e+FLw9oilevy29^M?5iYe;r8bNkPq8d788L2t7U
zO(djVNrV(~@_)p_V)<$E@y#j88``N~_yjUcKkl8~U>H5Wywhi=1W`z*cOMVWC~Cy(
zzz#J(Z;@U!b7k24KwG8_5jQJ`)BT~c@G>9^bnI=Yq3J&xdXY9GUNxqD>Djh$IFCgi4|1COvNjqQritichtS-&oXyZo=G$^`Xo+Bl1s#I7)5I=Tp>TkRvwiVn
zReiWiT8Tr6wWnWu1dRvjkw^<|rTQ?VkXn4XO+U%?8za(_{$Dzls*&e?7m9iC1$0hG
zUj0P^chm0^wfx#<(gq
zV$Hc_H{r&Hpa5{Dm1Sx1Ki`Nu(=Mcrx$j2wSIAaO4PBhKlibgkHN
z#c*QnXQ1K?pw;1FtBKq>7WGDV_deg`!c42AWxbmgI~aC2a3IhRyRS;Z$x%&YXZ_R4
zGsTl&$YKnkeHp0;mwO3g^nm}>Ou6=OO;Vy`uf4o}58>Woz51t5MR&Is>YV|5rg&7*
z@^iaCIvW>NO$taIFVvrdL1q+3bwz6H52T8TsLPmT<_r3O%$+=Nqk1u;;B=JtHS^D)
zV;6LtE%~FbR_a2zvNP`@wIc-NH|Lebaed6AHJUi#=1)f_6=G}EGEH4e=;C73KQ@8X
zu&u=}-?Rzp$0qF}_V;%X2GJR99ISF`4li+v?>A%AW^jJ?zg8ihEhPmFT43I2Psngq
z-6CAwne$FXlp;=t9<1Nzx%?rrT57+Yy9zUyEWlC29)6~19q69qSFU~4?-HcNal_|E
zZ5u}=93Pd!hW9a^x7BJAUs*Zf$Ky6FU1SSb+ULfxt+4Q;yi&Flm63*L$*vytmHPnp
z@?^3WGvq3L$LLySD6>mHn@S<>td{kNn!mnLqB)Qg&O#!ivE(-H4>LjTl{q4GJh*Cx
z>sn+9*TZ-7u9Qjg6;zpzp9mo$+4H`H_bnR^%>PhgyOX(VdYVtTJm9Umc*b`G@Qo;102x)4Tp}zM!)zc&CJnT(R;*)WDSG~1gtW%4E
zN}U-{g70j?h4@NdqtMC|I~T)-hJf=0LZVG=y_6K?7}CA1N_VH0P^WUZxZC&;XV5T?
zQO)&d0FZVSwG#pVdPuH(Ciz%t2;5q#B!ZqFr8rMeBJQ2S)fL@+HFHWN8XGQM{K5^b
zpQP93^E;T(+}HDb`z1wKrpYHQ=#1~NOOo&flm;SV=a-AdTrq}5;IhD+oX5&*p>XyS
zgJN%ye4Vz8Z0V8@OcwRcUuu2js&!#5y=Kb-ql4k+!*Q
From cbb35f683562246360c28529b1772098deddfcab Mon Sep 17 00:00:00 2001
From: memoryunreal <814514103@qq.com>
Date: Thu, 27 Apr 2023 07:56:26 +0000
Subject: [PATCH 2/7] for beta to merge -- li
---
app.py | 17 ++++++++++-------
1 file changed, 10 insertions(+), 7 deletions(-)
diff --git a/app.py b/app.py
index 5b66e92..0118220 100644
--- a/app.py
+++ b/app.py
@@ -13,10 +13,9 @@ import requests
import json
import torchvision
import torch
-from tools.interact_tools import SamControler
-from tracker.base_tracker import BaseTracker
from tools.painter import mask_painter
import psutil
+import time
try:
from mmcv.cnn import ConvModule
except:
@@ -82,7 +81,7 @@ def get_frames_from_video(video_input, video_state):
"""
video_path = video_input
frames = []
-
+ user_name = time.time()
operation_log = [("",""),("Upload video already. Try click the image for adding targets to track and inpaint.","Normal")]
try:
cap = cv2.VideoCapture(video_path)
@@ -103,6 +102,7 @@ def get_frames_from_video(video_input, video_state):
image_size = (frames[0].shape[0],frames[0].shape[1])
# initialize video_state
video_state = {
+ "user_name": user_name,
"video_name": os.path.split(video_path)[-1],
"origin_images": frames,
"painted_images": frames.copy(),
@@ -375,8 +375,8 @@ folder ="./checkpoints"
SAM_checkpoint = download_checkpoint(sam_checkpoint_url, folder, sam_checkpoint)
xmem_checkpoint = download_checkpoint(xmem_checkpoint_url, folder, xmem_checkpoint)
e2fgvi_checkpoint = download_checkpoint_from_google_drive(e2fgvi_checkpoint_id, folder, e2fgvi_checkpoint)
-# args.port = 12214
-# args.device = "cuda:2"
+# args.port = 12212
+# args.device = "cuda:1"
# args.mask_save = True
# initialize sam, xmem, e2fgvi models
@@ -409,6 +409,7 @@ with gr.Blocks() as iface:
video_state = gr.State(
{
+ "user_name": "",
"video_name": "",
"origin_images": None,
"painted_images": None,
@@ -532,6 +533,8 @@ with gr.Blocks() as iface:
video_input.clear(
lambda: (
{
+ "user_name": "",
+ "video_name": "",
"origin_images": None,
"painted_images": None,
"masks": None,
@@ -593,5 +596,5 @@ with gr.Blocks() as iface:
# cache_examples=True,
)
iface.queue(concurrency_count=1)
-# iface.launch(debug=True, enable_queue=True, server_port=args.port, server_name="0.0.0.0")
-iface.launch(debug=True, enable_queue=True)
\ No newline at end of file
+iface.launch(debug=True, enable_queue=True, server_port=args.port, server_name="0.0.0.0")
+# iface.launch(debug=True, enable_queue=True)
\ No newline at end of file
From 920e3a65fae59306a68267970daca70f24a047b9 Mon Sep 17 00:00:00 2001
From: Mingqi Gao