├── .gitignore ├── LICENSE ├── README.md ├── README_cn.md ├── assets ├── AI-logo.png ├── all_tabs_demo.gif ├── asr_demo.gif ├── chat_demo.gif ├── segmentation_demo.gif ├── tts_demo.gif ├── video_convertor_demo.gif ├── video_inpainter_demo.gif ├── visualchat_demo.jpg └── visualchat_demo.png ├── configs ├── asr_demo.yml ├── base.yml ├── chatbot_demo.yml ├── segmentation_demo.yml ├── tts_demo.yml ├── video_convertor_demo.yml ├── video_inpainter_demo.yml ├── visualchat_demo.yml └── webui_configs.yml ├── demo ├── bgm │ ├── Paris.mp3 │ ├── illusionary_daytime.mp3 │ ├── time_back.mp3 │ └── windy_hill.mp3 ├── fastsam │ └── examples │ │ ├── dogs.jpg │ │ ├── sa_10039.jpg │ │ ├── sa_11025.jpg │ │ ├── sa_1309.jpg │ │ ├── sa_192.jpg │ │ ├── sa_414.jpg │ │ ├── sa_561.jpg │ │ ├── sa_862.jpg │ │ └── sa_8776.jpg └── video_inpainter │ ├── test-sample0.mp4 │ ├── test-sample1.mp4 │ ├── test-sample2.mp4 │ ├── test-sample3.mp4 │ └── test-sample4.mp4 ├── model_weights └── chatglm │ └── chatglm2-6b-int4 │ ├── config.json │ ├── configuration_chatglm.py │ ├── modeling_chatglm.py │ ├── quantization.py │ ├── tokenization_chatglm.py │ └── tokenizer_config.json ├── models ├── RAFT │ ├── __init__.py │ ├── corr.py │ ├── datasets.py │ ├── demo.py │ ├── extractor.py │ ├── raft.py │ ├── update.py │ └── utils │ │ ├── __init__.py │ │ ├── augmentor.py │ │ ├── flow_viz.py │ │ ├── flow_viz_pt.py │ │ ├── frame_utils.py │ │ └── utils.py ├── __init__.py ├── misc.py ├── modules │ ├── base_module.py │ ├── deformconv.py │ ├── flow_comp_raft.py │ ├── flow_loss_utils.py │ ├── sparse_transformer.py │ └── spectral_norm.py ├── propainter.py ├── recurrent_flow_completion.py ├── sam │ ├── README.md │ ├── __init__.py │ ├── assets │ │ ├── masks1.png │ │ ├── masks2.jpg │ │ ├── model_diagram.png │ │ ├── notebook1.png │ │ └── notebook2.png │ ├── linter.sh │ ├── notebooks │ │ ├── automatic_mask_generator_example.ipynb │ │ ├── images │ │ │ ├── dog.jpg │ │ │ ├── groceries.jpg │ │ │ └── truck.jpg │ │ ├── onnx_model_example.ipynb │ │ └── predictor_example.ipynb │ ├── scripts │ │ ├── amg.py │ │ └── export_onnx_model.py │ ├── segment_anything │ │ ├── .DS_Store │ │ ├── __init__.py │ │ ├── automatic_mask_generator.py │ │ ├── build_sam.py │ │ ├── modeling │ │ │ ├── __init__.py │ │ │ ├── common.py │ │ │ ├── image_encoder.py │ │ │ ├── mask_decoder.py │ │ │ ├── prompt_encoder.py │ │ │ ├── sam.py │ │ │ └── transformer.py │ │ ├── predictor.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── amg.py │ │ │ ├── onnx.py │ │ │ └── transforms.py │ ├── setup.cfg │ └── setup.py └── tracker │ ├── base_tracker.py │ ├── config │ └── __init__.py │ ├── inference │ ├── __init__.py │ ├── image_feature_store.py │ ├── inference_core.py │ ├── kv_memory_store.py │ ├── memory_manager.py │ ├── object_info.py │ ├── object_manager.py │ └── utils │ │ ├── __init__.py │ │ ├── args_utils.py │ │ ├── burst_utils.py │ │ ├── frame_utils.py │ │ └── results_utils.py │ ├── model │ ├── __init__.py │ ├── aux_modules.py │ ├── big_modules.py │ ├── channel_attn.py │ ├── cutie.py │ ├── group_modules.py │ ├── losses.py │ ├── modules.py │ ├── transformer │ │ ├── __init__.py │ │ ├── object_summarizer.py │ │ ├── object_transformer.py │ │ ├── positional_encoding.py │ │ └── transformer_layers.py │ └── utils │ │ ├── __init__.py │ │ ├── memory_utils.py │ │ ├── parameter_groups.py │ │ └── resnet.py │ └── utils │ ├── __init__.py │ ├── image_saver.py │ ├── load_subset.py │ ├── log_integrator.py │ ├── logger.py │ ├── mask_mapper.py │ ├── palette.py │ ├── pano_utils.py │ ├── point_features.py │ ├── range_transform.py │ ├── tensor_utils.py │ └── time_estimator.py ├── requirements.txt ├── tools ├── __init__.py ├── ai_wrapper.py ├── base.py ├── chatglm_handler.py ├── chatvlm_handler.py ├── edgetts_handler.py ├── fastsam_handler.py ├── gpt_handler.py ├── inpainting_handler.py ├── sam_handler.py ├── tracker_handler.py ├── visualglm_handler.py └── whisper_handler.py ├── utils ├── __init__.py ├── chatglm_utils.py ├── fastsam │ ├── __init__.py │ ├── decoder.py │ ├── model.py │ ├── predict.py │ ├── prompt.py │ ├── tools.py │ └── utils.py ├── gradio_tabs │ ├── __init__.py │ ├── audio_tabs.py │ ├── chat_tabs.py │ ├── image_tabs.py │ └── video_tabs.py ├── gradio_utils.py ├── mask_painter.py ├── misc.py └── painter.py └── webui.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/README.md -------------------------------------------------------------------------------- /README_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/README_cn.md -------------------------------------------------------------------------------- /assets/AI-logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/AI-logo.png -------------------------------------------------------------------------------- /assets/all_tabs_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/all_tabs_demo.gif -------------------------------------------------------------------------------- /assets/asr_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/asr_demo.gif -------------------------------------------------------------------------------- /assets/chat_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/chat_demo.gif -------------------------------------------------------------------------------- /assets/segmentation_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/segmentation_demo.gif -------------------------------------------------------------------------------- /assets/tts_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/tts_demo.gif -------------------------------------------------------------------------------- /assets/video_convertor_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/video_convertor_demo.gif -------------------------------------------------------------------------------- /assets/video_inpainter_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/video_inpainter_demo.gif -------------------------------------------------------------------------------- /assets/visualchat_demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/visualchat_demo.jpg -------------------------------------------------------------------------------- /assets/visualchat_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/assets/visualchat_demo.png -------------------------------------------------------------------------------- /configs/asr_demo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/asr_demo.yml -------------------------------------------------------------------------------- /configs/base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/base.yml -------------------------------------------------------------------------------- /configs/chatbot_demo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/chatbot_demo.yml -------------------------------------------------------------------------------- /configs/segmentation_demo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/segmentation_demo.yml -------------------------------------------------------------------------------- /configs/tts_demo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/tts_demo.yml -------------------------------------------------------------------------------- /configs/video_convertor_demo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/video_convertor_demo.yml -------------------------------------------------------------------------------- /configs/video_inpainter_demo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/video_inpainter_demo.yml -------------------------------------------------------------------------------- /configs/visualchat_demo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/visualchat_demo.yml -------------------------------------------------------------------------------- /configs/webui_configs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/configs/webui_configs.yml -------------------------------------------------------------------------------- /demo/bgm/Paris.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/bgm/Paris.mp3 -------------------------------------------------------------------------------- /demo/bgm/illusionary_daytime.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/bgm/illusionary_daytime.mp3 -------------------------------------------------------------------------------- /demo/bgm/time_back.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/bgm/time_back.mp3 -------------------------------------------------------------------------------- /demo/bgm/windy_hill.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/bgm/windy_hill.mp3 -------------------------------------------------------------------------------- /demo/fastsam/examples/dogs.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/dogs.jpg -------------------------------------------------------------------------------- /demo/fastsam/examples/sa_10039.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/sa_10039.jpg -------------------------------------------------------------------------------- /demo/fastsam/examples/sa_11025.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/sa_11025.jpg -------------------------------------------------------------------------------- /demo/fastsam/examples/sa_1309.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/sa_1309.jpg -------------------------------------------------------------------------------- /demo/fastsam/examples/sa_192.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/sa_192.jpg -------------------------------------------------------------------------------- /demo/fastsam/examples/sa_414.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/sa_414.jpg -------------------------------------------------------------------------------- /demo/fastsam/examples/sa_561.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/sa_561.jpg -------------------------------------------------------------------------------- /demo/fastsam/examples/sa_862.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/sa_862.jpg -------------------------------------------------------------------------------- /demo/fastsam/examples/sa_8776.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/fastsam/examples/sa_8776.jpg -------------------------------------------------------------------------------- /demo/video_inpainter/test-sample0.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/video_inpainter/test-sample0.mp4 -------------------------------------------------------------------------------- /demo/video_inpainter/test-sample1.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/video_inpainter/test-sample1.mp4 -------------------------------------------------------------------------------- /demo/video_inpainter/test-sample2.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/video_inpainter/test-sample2.mp4 -------------------------------------------------------------------------------- /demo/video_inpainter/test-sample3.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/video_inpainter/test-sample3.mp4 -------------------------------------------------------------------------------- /demo/video_inpainter/test-sample4.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/demo/video_inpainter/test-sample4.mp4 -------------------------------------------------------------------------------- /model_weights/chatglm/chatglm2-6b-int4/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/model_weights/chatglm/chatglm2-6b-int4/config.json -------------------------------------------------------------------------------- /model_weights/chatglm/chatglm2-6b-int4/configuration_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/model_weights/chatglm/chatglm2-6b-int4/configuration_chatglm.py -------------------------------------------------------------------------------- /model_weights/chatglm/chatglm2-6b-int4/modeling_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/model_weights/chatglm/chatglm2-6b-int4/modeling_chatglm.py -------------------------------------------------------------------------------- /model_weights/chatglm/chatglm2-6b-int4/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/model_weights/chatglm/chatglm2-6b-int4/quantization.py -------------------------------------------------------------------------------- /model_weights/chatglm/chatglm2-6b-int4/tokenization_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/model_weights/chatglm/chatglm2-6b-int4/tokenization_chatglm.py -------------------------------------------------------------------------------- /model_weights/chatglm/chatglm2-6b-int4/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/model_weights/chatglm/chatglm2-6b-int4/tokenizer_config.json -------------------------------------------------------------------------------- /models/RAFT/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/__init__.py -------------------------------------------------------------------------------- /models/RAFT/corr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/corr.py -------------------------------------------------------------------------------- /models/RAFT/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/datasets.py -------------------------------------------------------------------------------- /models/RAFT/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/demo.py -------------------------------------------------------------------------------- /models/RAFT/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/extractor.py -------------------------------------------------------------------------------- /models/RAFT/raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/raft.py -------------------------------------------------------------------------------- /models/RAFT/update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/update.py -------------------------------------------------------------------------------- /models/RAFT/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/utils/__init__.py -------------------------------------------------------------------------------- /models/RAFT/utils/augmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/utils/augmentor.py -------------------------------------------------------------------------------- /models/RAFT/utils/flow_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/utils/flow_viz.py -------------------------------------------------------------------------------- /models/RAFT/utils/flow_viz_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/utils/flow_viz_pt.py -------------------------------------------------------------------------------- /models/RAFT/utils/frame_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/utils/frame_utils.py -------------------------------------------------------------------------------- /models/RAFT/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/RAFT/utils/utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/misc.py -------------------------------------------------------------------------------- /models/modules/base_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/modules/base_module.py -------------------------------------------------------------------------------- /models/modules/deformconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/modules/deformconv.py -------------------------------------------------------------------------------- /models/modules/flow_comp_raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/modules/flow_comp_raft.py -------------------------------------------------------------------------------- /models/modules/flow_loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/modules/flow_loss_utils.py -------------------------------------------------------------------------------- /models/modules/sparse_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/modules/sparse_transformer.py -------------------------------------------------------------------------------- /models/modules/spectral_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/modules/spectral_norm.py -------------------------------------------------------------------------------- /models/propainter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/propainter.py -------------------------------------------------------------------------------- /models/recurrent_flow_completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/recurrent_flow_completion.py -------------------------------------------------------------------------------- /models/sam/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/README.md -------------------------------------------------------------------------------- /models/sam/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/sam/assets/masks1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/assets/masks1.png -------------------------------------------------------------------------------- /models/sam/assets/masks2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/assets/masks2.jpg -------------------------------------------------------------------------------- /models/sam/assets/model_diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/assets/model_diagram.png -------------------------------------------------------------------------------- /models/sam/assets/notebook1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/assets/notebook1.png -------------------------------------------------------------------------------- /models/sam/assets/notebook2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/assets/notebook2.png -------------------------------------------------------------------------------- /models/sam/linter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/linter.sh -------------------------------------------------------------------------------- /models/sam/notebooks/automatic_mask_generator_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/notebooks/automatic_mask_generator_example.ipynb -------------------------------------------------------------------------------- /models/sam/notebooks/images/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/notebooks/images/dog.jpg -------------------------------------------------------------------------------- /models/sam/notebooks/images/groceries.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/notebooks/images/groceries.jpg -------------------------------------------------------------------------------- /models/sam/notebooks/images/truck.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/notebooks/images/truck.jpg -------------------------------------------------------------------------------- /models/sam/notebooks/onnx_model_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/notebooks/onnx_model_example.ipynb -------------------------------------------------------------------------------- /models/sam/notebooks/predictor_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/notebooks/predictor_example.ipynb -------------------------------------------------------------------------------- /models/sam/scripts/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/scripts/amg.py -------------------------------------------------------------------------------- /models/sam/scripts/export_onnx_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/scripts/export_onnx_model.py -------------------------------------------------------------------------------- /models/sam/segment_anything/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/.DS_Store -------------------------------------------------------------------------------- /models/sam/segment_anything/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/__init__.py -------------------------------------------------------------------------------- /models/sam/segment_anything/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/automatic_mask_generator.py -------------------------------------------------------------------------------- /models/sam/segment_anything/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/build_sam.py -------------------------------------------------------------------------------- /models/sam/segment_anything/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/modeling/__init__.py -------------------------------------------------------------------------------- /models/sam/segment_anything/modeling/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/modeling/common.py -------------------------------------------------------------------------------- /models/sam/segment_anything/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/modeling/image_encoder.py -------------------------------------------------------------------------------- /models/sam/segment_anything/modeling/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/modeling/mask_decoder.py -------------------------------------------------------------------------------- /models/sam/segment_anything/modeling/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/modeling/prompt_encoder.py -------------------------------------------------------------------------------- /models/sam/segment_anything/modeling/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/modeling/sam.py -------------------------------------------------------------------------------- /models/sam/segment_anything/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/modeling/transformer.py -------------------------------------------------------------------------------- /models/sam/segment_anything/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/predictor.py -------------------------------------------------------------------------------- /models/sam/segment_anything/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/utils/__init__.py -------------------------------------------------------------------------------- /models/sam/segment_anything/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/utils/amg.py -------------------------------------------------------------------------------- /models/sam/segment_anything/utils/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/utils/onnx.py -------------------------------------------------------------------------------- /models/sam/segment_anything/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/segment_anything/utils/transforms.py -------------------------------------------------------------------------------- /models/sam/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/setup.cfg -------------------------------------------------------------------------------- /models/sam/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/sam/setup.py -------------------------------------------------------------------------------- /models/tracker/base_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/base_tracker.py -------------------------------------------------------------------------------- /models/tracker/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/config/__init__.py -------------------------------------------------------------------------------- /models/tracker/inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/tracker/inference/image_feature_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/image_feature_store.py -------------------------------------------------------------------------------- /models/tracker/inference/inference_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/inference_core.py -------------------------------------------------------------------------------- /models/tracker/inference/kv_memory_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/kv_memory_store.py -------------------------------------------------------------------------------- /models/tracker/inference/memory_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/memory_manager.py -------------------------------------------------------------------------------- /models/tracker/inference/object_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/object_info.py -------------------------------------------------------------------------------- /models/tracker/inference/object_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/object_manager.py -------------------------------------------------------------------------------- /models/tracker/inference/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/tracker/inference/utils/args_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/utils/args_utils.py -------------------------------------------------------------------------------- /models/tracker/inference/utils/burst_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/utils/burst_utils.py -------------------------------------------------------------------------------- /models/tracker/inference/utils/frame_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/utils/frame_utils.py -------------------------------------------------------------------------------- /models/tracker/inference/utils/results_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/inference/utils/results_utils.py -------------------------------------------------------------------------------- /models/tracker/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/tracker/model/aux_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/aux_modules.py -------------------------------------------------------------------------------- /models/tracker/model/big_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/big_modules.py -------------------------------------------------------------------------------- /models/tracker/model/channel_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/channel_attn.py -------------------------------------------------------------------------------- /models/tracker/model/cutie.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/cutie.py -------------------------------------------------------------------------------- /models/tracker/model/group_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/group_modules.py -------------------------------------------------------------------------------- /models/tracker/model/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/losses.py -------------------------------------------------------------------------------- /models/tracker/model/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/modules.py -------------------------------------------------------------------------------- /models/tracker/model/transformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/tracker/model/transformer/object_summarizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/transformer/object_summarizer.py -------------------------------------------------------------------------------- /models/tracker/model/transformer/object_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/transformer/object_transformer.py -------------------------------------------------------------------------------- /models/tracker/model/transformer/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/transformer/positional_encoding.py -------------------------------------------------------------------------------- /models/tracker/model/transformer/transformer_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/transformer/transformer_layers.py -------------------------------------------------------------------------------- /models/tracker/model/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/tracker/model/utils/memory_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/utils/memory_utils.py -------------------------------------------------------------------------------- /models/tracker/model/utils/parameter_groups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/utils/parameter_groups.py -------------------------------------------------------------------------------- /models/tracker/model/utils/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/model/utils/resnet.py -------------------------------------------------------------------------------- /models/tracker/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/tracker/utils/image_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/image_saver.py -------------------------------------------------------------------------------- /models/tracker/utils/load_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/load_subset.py -------------------------------------------------------------------------------- /models/tracker/utils/log_integrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/log_integrator.py -------------------------------------------------------------------------------- /models/tracker/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/logger.py -------------------------------------------------------------------------------- /models/tracker/utils/mask_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/mask_mapper.py -------------------------------------------------------------------------------- /models/tracker/utils/palette.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/palette.py -------------------------------------------------------------------------------- /models/tracker/utils/pano_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/pano_utils.py -------------------------------------------------------------------------------- /models/tracker/utils/point_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/point_features.py -------------------------------------------------------------------------------- /models/tracker/utils/range_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/range_transform.py -------------------------------------------------------------------------------- /models/tracker/utils/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/tensor_utils.py -------------------------------------------------------------------------------- /models/tracker/utils/time_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/models/tracker/utils/time_estimator.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/ai_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/ai_wrapper.py -------------------------------------------------------------------------------- /tools/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/base.py -------------------------------------------------------------------------------- /tools/chatglm_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/chatglm_handler.py -------------------------------------------------------------------------------- /tools/chatvlm_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/chatvlm_handler.py -------------------------------------------------------------------------------- /tools/edgetts_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/edgetts_handler.py -------------------------------------------------------------------------------- /tools/fastsam_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/fastsam_handler.py -------------------------------------------------------------------------------- /tools/gpt_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/gpt_handler.py -------------------------------------------------------------------------------- /tools/inpainting_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/inpainting_handler.py -------------------------------------------------------------------------------- /tools/sam_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/sam_handler.py -------------------------------------------------------------------------------- /tools/tracker_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/tracker_handler.py -------------------------------------------------------------------------------- /tools/visualglm_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/visualglm_handler.py -------------------------------------------------------------------------------- /tools/whisper_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/tools/whisper_handler.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/chatglm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/chatglm_utils.py -------------------------------------------------------------------------------- /utils/fastsam/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/fastsam/__init__.py -------------------------------------------------------------------------------- /utils/fastsam/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/fastsam/decoder.py -------------------------------------------------------------------------------- /utils/fastsam/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/fastsam/model.py -------------------------------------------------------------------------------- /utils/fastsam/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/fastsam/predict.py -------------------------------------------------------------------------------- /utils/fastsam/prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/fastsam/prompt.py -------------------------------------------------------------------------------- /utils/fastsam/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/fastsam/tools.py -------------------------------------------------------------------------------- /utils/fastsam/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/fastsam/utils.py -------------------------------------------------------------------------------- /utils/gradio_tabs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/gradio_tabs/__init__.py -------------------------------------------------------------------------------- /utils/gradio_tabs/audio_tabs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/gradio_tabs/audio_tabs.py -------------------------------------------------------------------------------- /utils/gradio_tabs/chat_tabs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/gradio_tabs/chat_tabs.py -------------------------------------------------------------------------------- /utils/gradio_tabs/image_tabs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/gradio_tabs/image_tabs.py -------------------------------------------------------------------------------- /utils/gradio_tabs/video_tabs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/gradio_tabs/video_tabs.py -------------------------------------------------------------------------------- /utils/gradio_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/gradio_utils.py -------------------------------------------------------------------------------- /utils/mask_painter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/mask_painter.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/painter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/utils/painter.py -------------------------------------------------------------------------------- /webui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jasonaidm/ai_webui/HEAD/webui.py --------------------------------------------------------------------------------