如何安裝streamlit框架

這篇文章主要介紹“如何安裝streamlit框架”，在日常操作中，相信很多人在如何安裝streamlit框架問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”如何安裝streamlit框架”的疑惑有所幫助！接下來，請跟著小編一起來學(xué)習(xí)吧！

框架安裝

streamlit框架的安裝非常簡單，使用pip就可以安裝：

pip install streamlit

安裝完成之后，可以使用命令進(jìn)行版本驗(yàn)證：

streamlit --version

在這篇文章當(dāng)中，我會用一個(gè)實(shí)際工作中的例子簡單介紹streamlit框架的使用流程。

項(xiàng)目準(zhǔn)備

Collaborative-Distillation是一個(gè)支持高分辨率的圖像風(fēng)格化方案，該模型的輸入是風(fēng)格圖片以及待處理的圖片，輸出是風(fēng)格化之后的圖片。

在這個(gè)代碼倉庫當(dāng)中，我們需要使用比較復(fù)雜的命令行命令來進(jìn)行風(fēng)格化操作：

# use original VGG-19, normal images
CUDA_VISIBLE_DEVICES=0 python WCT.py --debug --mode original

# use original VGG-19, ultra-res images
CUDA_VISIBLE_DEVICES=0 python WCT.py --debug --mode original --UHD

# use our pruned VGG-19, normal images
CUDA_VISIBLE_DEVICES=0 python WCT.py --debug --mode 16x

# use our pruned VGG-19, ultra-res images
CUDA_VISIBLE_DEVICES=0 python WCT.py --debug --mode 16x --UHD

# If your RAM cannot afford some large images, you can change the content and style size via '--content_size' and '--style_size'
CUDA_VISIBLE_DEVICES=0 python WCT.py --debug --mode 16x --UHD --content_size 3000 --style_size 2000

但是這樣的操作對于用戶來說相當(dāng)復(fù)雜，所以我們可以使用streamlit編寫一個(gè)demo頁面，方便用戶使用。

在這個(gè)demo頁面當(dāng)中，會用到streamlit的以下幾種組件：

• streamlit.file_uploader：文件上傳組件，具體見下圖

  

  該組件支持拖拽上傳文件和文件管理器選擇文件，相對來說比較方便，使用方法如下列代碼所示：

  style_file = st.file_uploader("請上傳風(fēng)格化圖片")
  
  if style_file:
      stringio = style_file.getvalue()
      style_file_path = 'style_file/'+ style_file.name
      with open(style_file_path,'wb') as f:
          f.write(stringio)

  
  

  使用文件上傳組件上傳文件之后，可以使用上面的代碼將文件保存到特定路徑等待使用。

• streamlit.image：圖片顯示組件，具體見下圖：

  

  該組件可以在demo頁面中根據(jù)圖片路徑顯示圖片。

  style_file_path = 'style_file/'+ style_file.name
  st.image(style_file_path)

  
  

• streamlit.write：文字顯示組件，該組件可以在網(wǎng)頁上顯示一些提示信息。

  

  st.write('高分辨率風(fēng)格化demo')

  
  

• streamlit.button：按鈕組件，點(diǎn)擊之后可以進(jìn)行一些任務(wù)。

  

  if st.button('開始進(jìn)行風(fēng)格化處理'):
  	style_func()

  
  

• streamlit.progress：進(jìn)度顯示組件，可以用來顯示任務(wù)的進(jìn)度。

  

  for i in range(0,100,10):
  	st.progress(i + 1)

  
  

streamlit中還有一些重要的組件，例如：

• streamlit.cache：數(shù)據(jù)緩存組件，該組件可以作為裝飾器使用，用處是緩存數(shù)據(jù)，加快數(shù)據(jù)載入速度?？梢杂迷谛枰磸?fù)加載數(shù)據(jù)或者進(jìn)行計(jì)算的函數(shù)當(dāng)中。

  @st.cache
  def load_dataset(data_link):
      dataset = pd.read_csv(data_link)
      return dataset

  
  

• streamlit.audio：音頻展示組件，可以根據(jù)音頻地址播放音頻。

with open('audio.mp3','rb') as f:
	st.audio(f,format="audio/mp3")

• streamlit.audio：選擇組件，該組件可以讓用戶從多個(gè)選項(xiàng)中選擇一項(xiàng)。

  

model_choose = st.radio('請選擇分離模型：',['人聲+伴奏','人聲+鋼琴+吉他+鼓+其他'],0)

其中參數(shù)0表示默認(rèn)選擇第一項(xiàng)。

streamlit支持的組件還是很多的，如果感興趣，請參考官方文檔。

項(xiàng)目編寫

這個(gè)demo頁面的主要功能是讓用戶分別上傳style圖片和content圖片，然后后臺進(jìn)行風(fēng)格化操作，風(fēng)格化操作完成之后顯示結(jié)果圖片。這樣用戶就可以快速的進(jìn)行風(fēng)格化操作并知道結(jié)果。

streamlit應(yīng)用是用python語言編寫的。在python文件開頭，需要導(dǎo)入streamlit包。

import streamlit as st

接著進(jìn)行文件的上傳與預(yù)處理：

style_file = st.file_uploader("請上傳風(fēng)格化圖片")
content_file = st.file_uploader("請上傳待處理圖片")
image_slot = st.empty()

if style_file:
    stringio = style_file.getvalue()
    style_file_path = 'style_file/'+ style_file.name
    with open(style_file_path,'wb') as f:
        f.write(stringio)
    image_slot.image(style_file_path)

if content_file:
    stringio = content_file.getvalue()
    content_file_path = 'content_file/'+ content_file.name
    with open(content_file_path,'wb') as f:
        f.write(stringio)

if content_file and style_file:
    img1 = Image.open( style_file_path)
    img1 = img1.resize((640, 640))

    img2 = Image.open( content_file_path)
    img2 = img2.resize((640, 640))
    new_img = Image.new('RGB', (1280, 640), 255)
    new_img.paste(img1,(0,0))
    new_img.paste(img2,(640,0))
    new_img.save('concrate_file/' + os.path.basename(style_file_path))
    image_slot.image('concrate_file/' + os.path.basename(style_file_path))

最后寫一個(gè)按鈕，執(zhí)行風(fēng)格化操作，并顯示最終結(jié)果，同時(shí)添加一個(gè)進(jìn)度條：

if st.button('開始進(jìn)行風(fēng)格化處理'):
    my_bar = st.progress(10)

    UHD_content_folder_path = 'PytorchWCT/content/UHD_content'
    output_path = WCT_func.process(content_file_path,style_file_path)
    for i in range(0,100,10):
        my_bar.progress(i + 1)
    my_bar.progress(100)
    st.write('風(fēng)格化之后的圖片')
    st.image(output_path)

項(xiàng)目的運(yùn)行和部署

streamlit框架的運(yùn)行方式非常簡單，直接在命令行執(zhí)行：

$ streamlit run streamlit_demo.py

就可以在瀏覽器中進(jìn)行訪問了。

總結(jié)

streamlit框架非常適合快速編寫流程不太復(fù)雜且需要可視化操作的demo，作者從開始編寫到編寫完成這個(gè)demo用時(shí)不到半個(gè)小時(shí)，編寫代碼不到50行，而且運(yùn)行部署起來非常方便，頁面看起來要比使用flask之類的框架渲染出的網(wǎng)頁美觀許多，實(shí)乃算法工程師的利器。

到此，關(guān)于“如何安裝streamlit框架”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識，請繼續(xù)關(guān)注億速云網(wǎng)站，小編會繼續(xù)努力為大家?guī)砀鄬?shí)用的文章！