Skip to content

Offliners/RobotLab-MLDL-Training-2022

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

meme

NTUME Robot Lab - ML/DL Training

2022/07/23 released

Environment

  • Python 3.8.13
  • CUDA 11.3
  • Pytorch 1.11

Usage (run on local machine)

# 使用Anaconda建立名為training的虛擬環境
$ conda create --name training python=3.8.13

# 啟動名為training的虛擬環境
$ conda activate training

$ git clone https://github.com/Offliners/RobotLab-MLDL-Training-2022.git
$ cd RobotLab-MLDL-Training-2022

# 安裝此專案所需的所有函式庫
$ pip install -r requirements.txt

# 前往想要訓練的範例
$ cd tutorial-x

# 完成訓練後即可關閉虛擬環境
$ conda deactivate

# 若之後確定不會使用的話,可以移除虛擬環境
$ conda env remove -n training

Useful Tips for Google Colab

Prevent Google Colab from disconnecting (valid utill 2022/07/23)

Press F12,and enter this code in console,then press enter

function ClickConnect(){
  console.log("Connnect Clicked - Start"); 
  document.querySelector("#top-toolbar > colab-connect-button").shadowRoot.querySelector("#connect").click();
  console.log("Connnect Clicked - End"); 
};
setInterval(ClickConnect, 60000)

Auto save output file on Google Colab

insert code cell at the bottom

from google.colab import files
files.download("output_file.csv")  # "output_file.csv" must be your output file name

Display information of GPU of Google Colab

insert code cell to check which GPU is assigned

!nvidia-smi

"Sorry, something went wrong. Reload?" when viewing *.ipynb on Github

Copy the URL to https://nbviewer.jupyter.org/

Tutorial 1 - Covid19 Cases Prediction

透過相關係數的分析,選出重要的特徵來訓練模型做Covid19 Cases Prediction

demo

Dataset Resource : Covid19 Cases

Links : [README] [Jupyter Notebook] [Google Colab]

Tutorial 2 - Food Classification

透過Teacher-Student的架構,訓練出3個學生模型,並使用Ensemble與Test Time Augmentation的技術來完成Food Classification

demo

Dataset Resource : Food11

Links : [README] [Jupyter Notebook] [Google Colab]

Tutorial 3 - Synthetic Image Segmentation

使用resnet18當backbone,並用UNet來完成Synthetic Image Segmentation

demo

Links : [README] [Jupyter Notebook] [Google Colab]

Tutorial 4 - Anime Face Generation

使用SNGAN來完成Anime Face Generation

demo

Dataset Resource : Crypko

Links : [README] [Jupyter Notebook] [Google Colab]

Tutorial 5 - Super Mario

使用PPO策略來訓練模型完成Super Mario任務

World/Stage 1 2 3 4
1 World 1-1 World 1-2 World 1-3 World 1-4
2 World 2-1 World 2-2 World 2-3 World 2-4
3 World 3-1 World 3-2 World 3-3 World 3-4
4 World 4-1 World 4-2 World 4-3 World 4-4
5 World 5-1 World 5-2 World 5-3 World 5-4
6 World 6-1 World 6-2 World 6-3 World 6-4
7 World 7-1 World 7-2 World 7-3 World 7-4
8 World 8-1 World 8-2 World 8-3 World 8-4

Environment : gym-super-mario-bros

Links : [README] [Jupyter Notebook] [Google Colab]

References

About

NTUME Robot Lab ML/DL Training (date : 2022/07/23)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published