One Stage Detector

• CenterNet
• FaceDetector and Recognition
  • Face Detector based on CenterNet
  • Face Detector based on CenterNet FPN
  • Face Net
  • Face Net with ArcFace Loss
• YoloV3
• GaussianYoloV3

Segmentation

• Segmentation

Classification

• Classification

Development environment

• OS : ubuntu linux 18.04 LTS
• Graphic card / driver : Quadro RTX 5000 / 470.103.01
• miniconda version : 4.11.0
• pytorch version : 1.11.0
  • Configure Run Environment
    1. Create a virtual environment

    jg@JG:~$ conda create -n pytorch python==3.8.0

    2. Install the required module

    jg@JG:~$ conda activate pytorch 
    (pytorch) jg@JG:~$ conda install pytorch torchvision cudatoolkit cudnn -c pytorch
    (pytorch) jg@JG:~$ pip install matplotlib tensorboard torchsummary plotly mlflow opencv-python tqdm PyYAML --no-cache-dir --pre --upgrade
    
    to do : torchsummary -> torchinfo 코드 
    to do : hiddenlayer -> build_graph 코드 작성

Author

• medical18@naver.com / JONGGON