Mirrors/Track-Anything
1
0
Fork 0
mirror of https://github.com/gaomingqi/Track-Anything.git synced 2025-12-16 08:27:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

update mmcv install from app.py -- li

Browse Source
This commit is contained in:
memoryunreal
2023-04-26 02:11:25 +00:00
parent 242f69052d
commit 1dcb4d91c7
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app.py
View File

@@ -14,6 +14,10 @@ import json
import torchvision import torchvision
import torch import torch
from tools.painter import mask_painter from tools.painter import mask_painter
try:
from mmcv.cnn import ConvModule
except:
os.system("mim install mmcv")
# download checkpoints # download checkpoints
def download_checkpoint(url, folder, filename): def download_checkpoint(url, folder, filename):
@@ -365,7 +369,7 @@ with gr.Blocks() as iface:
video_input = gr.Video(autosize=True) video_input = gr.Video(autosize=True)
with gr.Column(): with gr.Column():
video_info = gr.Textbox() video_info = gr.Textbox()
video_info = gr.Textbox(value="If you want to use the inpaint function, it is best to download and use a machine with more VRAM locally. \ resize_info = gr.Textbox(value="If you want to use the inpaint function, it is best to download and use a machine with more VRAM locally. \
Alternatively, you can use the resize ratio slider to scale down the original image to around 360P resolution for faster processing.") Alternatively, you can use the resize ratio slider to scale down the original image to around 360P resolution for faster processing.")
resize_ratio_slider = gr.Slider(minimum=0.02, maximum=1, step=0.02, value=1, label="Resize ratio", visible=True) resize_ratio_slider = gr.Slider(minimum=0.02, maximum=1, step=0.02, value=1, label="Resize ratio", visible=True)

3
requirements.txt
View File

@@ -15,5 +15,4 @@ onnx
metaseg==0.6.1 metaseg==0.6.1
pyyaml pyyaml
av av
mmcv-full openmim
mmengine