mooch443 / trex

TRex, a fast multi-animal tracking system with markerless identification, and 2D estimation of posture and visual fields.
https://trex.run
GNU General Public License v3.0
81 stars 8 forks source link

Training images not being saved #196

Closed mi2celis closed 1 year ago

mi2celis commented 1 year ago

Describe the bug Using trex for tracking a video. I am using auto_train and trying to save the training images by using recognition_save_training_images = true Training completes successfully, however, the training images are not being saved.

My settings for trex are as follows:

blob_size_ranges = [[0.0170333,3.089644]]
track_threshold = 15
track_max_individuals = 10
track_max_speed = 200
output_posture_data=true
auto_train = true
recognition_image_scale = 0.75
recognition_image_size = [256,256]
recognition_save_training_images = true

and the command I use is

trex \
    -d ${EWD}/results/${BASE} \
    -i ${EWD}/videos_pv/${BASE}.pv \
    -s ${PARAMS} \
    -log_file track_${BASE}.log

where EWD, BASE and PARAMS are defined. All other results are being saved to ${EWD}/results/${BASE}.

TRex is running on Ubuntu 20.04.4 LTS. Have NVIDIA Corporation TU106M [GeForce RTX 2070 Mobile] / Mesa Intel® UHD Graphics 630 (CFL GT2) TRex v1.1.9

Log file from trex session is attached output.txt

mooch443 commented 1 year ago

So, looking at your log, have you checked"/home/mimi/data/lab/trex/experiments/EXP_20211208153624//results/20211208-153646___G_pZMafXy___F_008___D_001___A_180___O_500___S_700/20211208-153646___G_pZMafXy___F_008___D_001___A_180___O_500___S_700_L_validation_data.npz"?

mi2celis commented 1 year ago

I misunderstood. I thought the images would be saved as png files of something similar.

I see that validation_data.npz has ['validation_ids', 'validation_images', 'ids', 'images']

I will write some code to extract the images.

Thank you