Improve compatibility and documentation (analogdevicesinc#283)

README.md

@@ -1,6 +1,6 @@
 # ADI MAX78000/MAX78002 Model Training and Synthesis
 
-December 22, 2023
+January 10, 2024
 
 ADI’s MAX78000/MAX78002 project is comprised of five repositories:
 
@@ -1534,6 +1534,11 @@ The following table describes the most important command line arguments for `tra
 | `--nas`                    | Enable network architecture search                           |                                 |
 | `--nas-policy`             | Define NAS policy in YAML file                               | `--nas-policy nas/nas_policy.yaml` |
 | `--regression` | Select regression instead of classification (changes Loss function, and log output) |  |
+| `--dr` | Set target embedding dimensionality for dimensionality reduction                |`--dr 64`                        |
+| `--scaf-lr` | Initial learning rate for sub-center ArcFace loss optimizer |  |
+| `--scaf-scale` |Scale hyperparameter for sub-center ArcFace loss |  |
+| `--scaf-margin` |Margin hyperparameter for sub-center ArcFace loss |  |
+| `--backbone-checkpoint` |Path to checkpoint from which to load backbone weights |  |
 | *Display and statistics*   |                                                              |                                 |
 | `--enable-tensorboard`     | Enable logging to TensorBoard (default: disabled)            |                                 |
 | `--confusion`              | Display the confusion matrix                                 |                                 |
@@ -1552,6 +1557,7 @@ The following table describes the most important command line arguments for `tra
 | `--summary onnx_simplified` | Export trained model to simplified [ONNX](https://onnx.ai/) file (default name: model.onnx) |                     |
 | `--summary-filename`       | Change the file name for the exported model                  | `--summary-filename mnist.onnx` |
 | `--save-sample`            | Save data[index] from the test set to a NumPy pickle for use as sample data | `--save-sample 10` |
+| `--slice-sample`  | For models that require RGB input, when the sample from the dataset has additional channels, slice the sample into 3 channels                                      |                                 |
 
 #### ONNX Model Export
 

requirements.txt

@@ -7,14 +7,14 @@ PyYAML>=5.1.1
 albumentations>=1.3.0
 faiss-cpu==1.7.4
 h5py>=3.7.0
-hawk-eyes==2.1.2
+hawk-eyes==2.1.0
 imutils==0.5.4
 kornia==0.6.8
 librosa>=0.7.2
 numba<0.50.0
 numpy>=1.22,<1.23
 onnx==1.15.0
-onnxruntime-gpu==1.10.0
+onnxruntime==1.7.0
 opencv-python>=4.4.0
 protobuf>=3.20.1,<4.0
 pycocotools==2.0.7