提交 8062ca40 编写于 作者: C chenguowei01

update dataset.py

上级 b4c2024c
......@@ -31,8 +31,9 @@ class EndSignal():
def is_pic(img_name):
valid_suffix = ['JPEG', 'jpeg', 'JPG', 'jpg', 'BMP', 'bmp', 'PNG', 'png']
valid_suffix = ['jpeg', 'jpg', 'bmp', 'png']
suffix = img_name.split('.')[-1]
suffix = suffix.lower()
if suffix not in valid_suffix:
return False
return True
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册