NISE: A novel approach by Piyavach K., piyavach.k@camt.info and Waranya M., earthmahanan@gmail.com
Proposed in IEA/AIE 2025
Last updated: August 7, 2025
Download trained weights, datasets, and related files from Google Drive.
Open Google Drivestrip_pre.py. The code contains step-by-step comments (steps 0 to
3)
explaining how to create the NISE.pip.strip_pre.pylib.pyoutput_k_fold, output_k_fold2,
output_k_fold3 which produce ready to use for training, validating and
testing
for YOLOv5 format.
A more efficient Numerical-to-Image learning pipeline for feature-based classification using YOLO.
Published in the AIAI / EANN–EAAAI 2026 Conference Proceedings.
The paper is available online through SpringerLink. The NISEv2 source code and requirements.txt are available on Google Drive.
Automated preprocessing, k-fold generation, and multi-dataset training through a single runner script.
Designed to support more datasets and larger-scale experiments than NISEv1.
Standardized project folders, backups, generated outputs, and repeatable Windows installation steps.
The following setup was tested with Python 3.10, PyTorch 2.7.1, CUDA 12.8, and an NVIDIA RTX 5050 Laptop GPU.
sm_120.
cd C:\yolo8_strip
py -3.10 -m venv venv
venv\Scripts\activate
python -m pip install --upgrade pip setuptools wheel
python -m pip --version
python -m pip install typing-extensions==4.12.2
python -m pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu128
python -c "import torch; print('Torch:', torch.__version__); print('CUDA:', torch.version.cuda); print('Available:', torch.cuda.is_available()); print('GPU:', torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU')"
A successful installation should report PyTorch 2.7.1+cu128, CUDA 12.8, GPU availability as True, and the detected NVIDIA GPU name.
python -m pip install -r requirements.txt
requirements.txt should not include the old CUDA 11.8 PyTorch entries, the local editable path from another computer, or the legacy Git-based pycocotools entry unless it is actually required.
C:\yolo8_strip\ultralytics
│
├─ run_all_datasets.py
├─ strip_pre2.py
├─ run_plain_all.py
│
├─ wine.csv
├─ iris.csv
├─ breast-cancer-wisconsin1.csv
├─ Student_performance_data.csv
├─ adult.csv
│
└─ backup
├─ wine
│ ├─ output_kfold_data.json
│ └─ output_strip_label_map.txt
├─ iris
├─ adult
└─ ...
Activate the virtual environment and enter the Ultralytics project folder:
cd C:\yolo8_strip
venv\Scripts\activate
cd C:\yolo8_strip\ultralytics
python run_all_datasets.py --datasets iris --epochs 1000 --batch 2 --device 0 --clean_each_project --exist_ok
python run_all_datasets.py --datasets iris --epochs 1 --batch 2 --device 0 --clean_each_project --exist_ok
python run_all_datasets.py --datasets wine,iris,breast_cancer,student,adult --epochs 1000 --batch 2 --device 0 --clean_each_project --exist_ok
--clean_each_project removes the current root-level generated k-fold and label-map files so that they are regenerated for the selected dataset.
| Dataset argument | Expected CSV filename |
|---|---|
wine | wine.csv |
iris | iris.csv |
breast_cancer | breast-cancer-wisconsin1.csv |
student | Student_performance_data.csv |
adult | adult.csv |
Last updated: July 18, 2026