Fixed demo.py to work with non GPU-enabled computers

scaelles / DEXTR-PyTorch

Deep Extreme Cut http://www.vision.ee.ethz.ch/~cvlsegmentation/dextr

GNU General Public License v3.0

843 stars 153 forks source link

Closed rawmarshmellows closed 6 years ago

rawmarshmellows commented 6 years ago

The script didn't work with my Macbook which didn't have a GPU so I added an use_cuda flag :)

scaelles commented 6 years ago

Thanks a lot for your suggestion, I have updated the demo with a simplified version of your code.