Skip to content

Instantly share code, notes, and snippets.

View mzr1996's full-sized avatar

Ma Zerun mzr1996

  • OpenMMLab
  • Shanghai
View GitHub Profile
@mzr1996
mzr1996 / make_multi_task_demo.py
Last active April 29, 2022 10:25
Create an demo multi-task dataset from CIFAR10
from pathlib import Path
import mmcv
from mmcls.datasets import build_dataset
data_root = Path("./cifar10")
extract_folder = data_root / 'images'
extract_folder.mkdir(parents=True, exist_ok=True)
dataset_cfg = dict(type='CIFAR10', data_prefix='cifar10', pipeline=())
dataset = build_dataset(dataset_cfg)
@mzr1996
mzr1996 / startup.sh
Created August 12, 2021 02:00
配置新环境
option() {
doc=$1
command=$2
echo "$doc ($command)"
while true; do
read -p "y(es)|n(o)|q(uit) " yn
case $yn in
[Yy]* ) eval $command; break;;
[Nn]* ) break;;
[Qq]* ) exit;;