Jupyter lab & Swin OD model Inference
최대 1 분 소요
jupyterlab
# venv activate
source venv/bin/activate
# install jupyter lab
pip install jupyter lab
# use config file - run jupyter lab
jupyter lab config=jupyter_notebook_config.py
Model Serving
- https://github.com/SwinTransformer/Swin-Transformer-Object-Detection
- 위 코드를 활용하여 inference 하고싶음!
- https://github.com/open-mmlab/mmdetection/blob/master/docs/get_started.md
# model file download
wget https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_mask_rcnn_swin_tiny_patch4_window7_1x.pth
# coco annotations download
wget http://images.cocodataset.org/annotations/annotations_trainval2017.zip
# unzip
unzip annotations_trainval2017.zip
# and make data/coco/annotation dir in root of Swin-Transformer dir
mkdir data
mkdir data/coco
cd data/coco
cp -r Swin-Transformer-Object-Detection/annotations .
# test inference execution : 근데 아직 에러남
python tools/test.py configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py mask_r_cnn_swinT_Lr1x/mask_rcnn_swin_tiny_patch4_window7_1x.pth --eval bbox segm
- 다 하고 위 inference command 수행했지만
- KeyError: “MaskRCNN: ‘SwinTransformer is not in the models registry’”
- 뭔가 설치가 덜 되었나..
댓글남기기