Remove opencv dependency and replace with torch logic.

Remove opencv dependency and replace with torch logic. #307