Test Yolact

紀錄一下步驟，ref:

• yolact 教學
• Install pytorch in annaconda

只是要test 一下 infer。

所以 create 一個 env，並且依照 ref:1 的說明安裝yolact 需要的 packages:

conda create --name yolact-3.7.9 python=3.7.9
conda activate yolact-3.7.9
pip install cypthon opencv-python pillow pycocotools matplotlib

然後 clone yolact project source:

git clone https://github.com/dbolya/yolact.git

然後download trained 好的 weights，在yolact 的 Evaluation 有download link。
選 yolact_resnet50_54_800000.pth
mkdir weights 並且 copy .pth 進去。

另外，安裝 pytorch，參考 ref:2

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

是 10.2，所以我要用 10.2

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

裝完測試一下：

python
>>import torch
>> torch.cuda.is_available()
True

OK. test infer:

python eval.py --trained_model=weights/yolact_resnet50_54_800000.pth --image=../test.jpg

source 的 save/load 只有做 weight 的部份。
所以要save model and weight 要自己改。
yolact.py : torch.save( )