COMS4995-Project

A Hybrid Ensembled Solution for SIIM-FISABIO-RSNA COVID-19 Detection

Environment setup

HARDWARE: 1 x NVIDIA GeForce 3090 24G GPU, 32 vCPUs, 78GB RAM
Ubuntu 18.04.6 LTS
CUDA 11.4
Python 3.9.12

$ sudo apt-get --assume-yes update
$ sudo apt-get --assume-yes install git man wget curl unzip zip fish htop tmux
$ sudo apt-get update -y && apt-get install software-properties-common -y && \
    add-apt-repository -y multiverse && apt-get update -y && apt-get upgrade -y && \
    apt-get install -y apt-utils nano vim man build-essential wget sudo && \
    rm -rf /var/lib/apt/lists/*
$ sudo apt-get --assume-yes install ffmpeg libsm6 libxext6  -y
$ sudo apt-get update

Dataset

Download processed SIIM dataset from google bucket:

$ echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] https://packages.cloud.google.com/apt cloud-sdk main" | sudo tee -a /etc/apt/sources.list.d/google-cloud-sdk.list
$ apt-get install apt-transport-https ca-certificates gnupg 
$ curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | sudo apt-key --keyring /usr/share/keyrings/cloud.google.gpg add -
$ apt-get update && sudo apt-get install google-cloud-sdk 
$ apt-get install google-cloud-sdk-app-engine-python --assume-yes 
$ gcloud init 
$ gsutil cp gs://siimcovid19_jpg/siim_512_jpg.zip .
$ unzip siim_512_jpg.zip

Getting Started

$ git clone https://github.com/psui3905/COMS4995-Project.git
$ mv jpg_form ~/COMS4995-Project
$ cd COMS4995-Project
$ pip install -r requirement.txt
$ python3 train.py

Contribution

This project is equally contributed by Pengwei Sui (ps3307) and Zheyu Zhang (zz2980).