libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)
https://ffcv.io
Apache License 2.0
2.79k stars 180 forks source link

[General question] FFCV scope #354

Open Arnold2381 opened 8 months ago

Arnold2381 commented 8 months ago

Hi team,

Does FFCV only works for image dataset? Can i use it for non image dataset too, like tabular data and so on.

Also in addition to that i am getting some installation failure, I used the dockerfile from the readme but when on installing and importing the packages, this is the exception i get.

2023-10-26 17:20:10 Traceback (most recent call last):
2023-10-26 17:20:10   File "./train.py", line 1, in <module>
2023-10-26 17:20:10     from ffcv.loader import Loader, OrderOption
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/ffcv/__init__.py", line 1, in <module>
2023-10-26 17:20:10     from .loader import Loader
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/ffcv/loader/__init__.py", line 1, in <module>
2023-10-26 17:20:10     from .loader import Loader, OrderOption
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/ffcv/loader/loader.py", line 14, in <module>
2023-10-26 17:20:10     from ffcv.fields.base import Field
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/ffcv/fields/__init__.py", line 3, in <module>
2023-10-26 17:20:10     from .rgb_image import RGBImageField
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/ffcv/fields/rgb_image.py", line 5, in <module>
2023-10-26 17:20:10     import cv2
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/cv2/__init__.py", line 181, in <module>
2023-10-26 17:20:10     bootstrap()
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/cv2/__init__.py", line 175, in bootstrap
2023-10-26 17:20:10     if __load_extra_py_code_for_module("cv2", submodule, DEBUG):
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/cv2/__init__.py", line 28, in __load_extra_py_code_for_module
2023-10-26 17:20:10     py_module = importlib.import_module(module_name)
2023-10-26 17:20:10   File "/usr/lib/python3.8/importlib/__init__.py", line 127, in import_module
2023-10-26 17:20:10     return _bootstrap._gcd_import(name[level:], package, level)
2023-10-26 17:20:10   File "/usr/local/lib/python3.8/dist-packages/cv2/typing/__init__.py", line 169, in <module>
2023-10-26 17:20:10     LayerId = cv2.dnn.DictValue
2023-10-26 17:20:10 AttributeError: module 'cv2.dnn' has no attribute 'DictValue'