試玩~Fully Convolutional Networks(PyTorch)

動機:想要玩玩 FCN 程式(PyTorch),可以嗎?!

準備環境
1.Ubuntu (GTX 1080 Ti * 2)
2.Python 3.7
3.PyTorch 0.4.0

實作步驟
1.作者(Kentaro Wada)分享了範例碼(如~參攷1.),下載安裝如下
$ git clone https://github.com/wkentaro/pytorch-fcn.git
$ cd pytorch-fcn
$ pip3 install torchfcn
2. 接著要 Training...如下
$ cd examples/voc
$ ./download_dataset.sh
其中有一個檔案下載有問題,要另外到 Google Drive 下載...如下
https://drive.google.com/file/d/1EQSKo5n2obj7tW8RytYTJ-eEYbXqtUXE/view?usp=sharing
下載的檔名為 benchmark.tgz 要改為 benchmark.tar 再解壓縮 $ tar -xvf benchmark.tar 才行...

$ python3 train_fcn32s.py -g 1
$ python3 train_fcn16s.py -g 1
$ python3 train_fcn8s.py -g 1
$ python3 train_fcn8s_atonce.py -g 1
執行上面的程式,每一隻都要用GPU跑 4~5 小時...我已累了!!

心得:等 training 完成後,再來 testing 吧...看來 FCN 不是我這個笨蛋想得這麼簡單!!!

參攷
1.wkentaro/pytorch-fcn, https://github.com/wkentaro/pytorch-fcn

留言