🛠 Toolbox to extend PyTorch functionalities
-
Updated
May 13, 2024 - Python
🛠 Toolbox to extend PyTorch functionalities
将原本Keras版本的AdvancedEAST改写成PyTorch版,将数据集由.npy文件改成一个LMDB文件,加入Precision,Recall, F1 score方便训练以及调试,底层网络仍然用VGG16。
A simple Lightning Memory-Mapped Database (LMDB) converter for ImageFolder datasets in PyTorch. Using LMDB over a regular file structure improves I/O performance significantly. Works on both Windows and Linux. Comes with latest Python support.
Add a description, image, and links to the lmdb-dataset topic page so that developers can more easily learn about it.
To associate your repository with the lmdb-dataset topic, visit your repo's landing page and select "manage topics."