test.py

@mbiparva 
how to use test.py? after I trained,   I tried python test.py -i 20200812_135411_361917 -p 039, but failed
└── 20200812_135411_361917
    ├── 004.pth
    ├── 009.pth
    ├── 014.pth
    ├── 019.pth
    ├── 024.pth
    ├── 029.pth
    ├── 034.pth
    └── 039.pth