search

videturfortuna / vehicle_reid_itsc2023

33 stars 5 forks source link

How to use the models? #4

Open tjasmin111 opened 11 months ago

tjasmin111 commented 11 months ago

Very interesting work! 2 basic questions:

  1. How can I test and run prediction on a single image using your pretrained weights?
  2. Is there a way to train a smaller model smaller? How can I do that? I see some parameters in main.py when passing a Module.nn.