aqeelanwar / MaskTheFace

Convert face dataset to masked dataset
http://www.masktheface.aqeel-anwar.com
MIT License
565 stars 148 forks source link

Handle filename contains multiple "." #23

Open ntvthuyen opened 2 years ago

ntvthuyen commented 2 years ago

split_path = f.rsplit(".") only can't handle filenames that contain more than "." Ex: file.30.26.jpg It will produce an opencv write error. (It will try to cv.imwrite("file.30",img ). In this modification, I join all elements of split_path array from the start to (n - 1) th to create new split_path with just 2 elements. Ex: file.30.26.jpg -> ["file","30","26","jpg"] -> ["file.30.26","jpg"]