📜 Paper   💻 Code   🌐 Project Page

The codes was tested on Windows 10, with Python and PyTorch. Required packages:

• numpy
• tqdm
• python
• matplotlib
• torch
• torchvision
• pandas
• opencv-python
• pyyaml

This repository contains codes for OAE(optical aberration estimation).

OAE
|   README.md
|   main.py
|   generate_edges_rgb.py
|   generate_fov_weight.py
|
|---configs
|   |   lensname.yaml
|
|---sfrmat5_dist
|   |---sfrmat5
|       |   user_sfrmat5_rgb.m
|
|---dataset 
|   |---lensname
|       |   npy
| 
|---input 
|   |   lensname.xlsx
| 
|---model 
|   |   optics_rgb.py
|   |   PSF_mlp.py
| 
|---results 
|
|---utils 
|   |   tools.py
|   |   train.py

/model contains the optical aberration model.

/dataset includes datasets used for training the optical aberration model.

/sfrmat5_dist contains the SFR calculation algorithm, which was downloaded from ISO 12233.

/results stores the results, including the PSF map and PSF comparisons.

Follow these steps to generate the necessary SFR (Spatial Frequency Response) data for training.

1. Generate Blurry Edges: Run the Python script generate_edges_rgb.py to process your images and obtain the initial blurry edge data.

   python generate_edges_rgb.py

2. Calculate SFR Curves: Next, open MATLAB and execute the user_sfrmat5_rgb.m script. This will analyze the edges from the previous step to compute and save the SFR curves.

3. Generate Training Data: Finally, run the generate_fov_weight.py script. This will process the SFR curves and generate the final .npz files that will be used as input for the training process.

   python generate_fov_weight.py

To train an aberration learning model from scratch, run main.py. The results will be saved in /results/lensname.

1. Why is F=5 used as the scaling factor (see scale in optics_rgb.py)?​​
When extracting a 2D MTF slice from the 3D MTF data (as implemented in slice within tool.py), we intentionally scale the coordinate system by a factor of 5.
This expansion:
- Reduces quantization errors during interpolation
- Improves numerical precision in the slicing operation
The scaling factor (F=5) is subsequently applied to compensate for this coordinate expansion, ensuring the final results maintain their proper scale and accuracy.

For any questions, please contact: meview.global@gmail.com