图片爬取+筛选

所需库

pip install umap-learn
pip install opencv-python
pip install matplotlib
pytorch

数据加载

修改相应文件的root_dir和file_suffix参数

运行

运行初级数据筛选：

python img_primary_filter.py

运行高级数据筛选：

python img_advanced_filter.py