README update for DELF training with autoencoder (#9607)
* Merged commit includes the following changes:
326369548 by Andre Araujo:
Fix import issues.
--
326159826 by Andre Araujo:
Changed the implementation of the cosine weights from Keras layer to tf.Variable to manually control for L2 normalization.
--
326139082 by Andre Araujo:
Support local feature matching using ratio test.
To allow for easily choosing which matching type to use, we rename a flag/argument and modify all related files to avoid breakages.
Also include a small change when computing nearest neighbors for geometric matching, to parallelize computation, which saves a little bit of time during execution (argument "n_jobs=-1").
--
326119848 by Andre Araujo:
Option to measure DELG latency taking binarization into account.
--
324316608 by Andre Araujo:
DELG global features training.
--
323693131 by Andre Araujo:
PY3 conversion for delf public lib.
--
321046157 by Andre Araujo:
Purely Google refactor
--
PiperOrigin-RevId: 326369548
* Added export of delg_model module.
* README update to explain training DELG global features head
* Added guidelines for DELF hyperparameter values
* Fixed typo
* Added mention about remaining training flags.
* Merged commit includes the following changes:
334723489 by Andre Araujo:
Backpropagate global and attention layers together.
--
334228310 by Andre Araujo:
Enable scaling of local feature locations to the resized resolution.
--
PiperOrigin-RevId: 334723489
* Merged commit includes the following changes:
347032253 by Andre Araujo:
Updated local and global_and_local model export scripts for exporting models trained with the autoencoder layer.
--
344312455 by Andre Araujo:
Implement autoencoder in training pipeline.
--
341116593 by Andre Araujo:
Reduce the default save_interval, to get more frequent checkpoints.
--
341111808 by Andre Araujo:
Allow checkpoint restoration in DELF training, to enable resuming of training jobs.
--
340138315 by Andre Araujo:
DELF training script: make it always save the last checkpoint.
--
338731551 by Andre Araujo:
Add image_size flag in DELF/G OSS training script.
--
338684879 by Andre Araujo:
Clean up summaries in DELF/G training script.
- Previously, the call to tf.summary.record_if() was not working, which led to summaries being recorded at every step, leading to too large events files. This is fixed.
- Previously, some summaries were computed at iteration k, while others at iteration k+1. Now, we standardize summary computations to always run after backpropagation (which means that summaries are reported for step k+1, referring to the batch k).
- Added a new summary: number of global steps per second; useful to see how fast training is making progress.
Also a few other small modifications are included:
- Improved description of the train.py script.
- Some small automatic reformattings.
--
PiperOrigin-RevId: 347032253
* Updated README for training with autoencoder
* DELF README update after first review.
Co-authored-by:
Andre Araujo <andrearaujo@google.com>
Showing
Please register or sign in to comment