search

Yiming-M / CLIP-EBC

The official implementation of the crowd counting model CLIP-EBC.
MIT License
35 stars 6 forks source link

Error in evaluation #14

Closed BurstLink666 closed 2 months ago

BurstLink666 commented 2 months ago

Hi, I got an error during the evaluation.

Traceback (most recent call last): File "/mnt/share/herenjie/CLIP-EBC/trainer.py", line 245, in main() File "/mnt/share/herenjie/CLIP-EBC/trainer.py", line 241, in main run(0, 1, args) File "/mnt/share/herenjie/CLIP-EBC/trainer.py", line 169, in run curr_val_scores = evaluate( File "/mnt/share/herenjie/CLIP-EBC/eval.py", line 35, in evaluate assert image_height % window_size == 0 and image_width % window_size == 0, f"Image size {image.shape} should be divisible by window size {window_size}." AssertionError: Image size torch.Size([1, 3, 704, 1024]) should be divisible by window size 448.

It seems that some images in val datasets cannot be divided by 448, which is the default window_size as you provided after executing preprocess.sh. I got this error when training clip_resnet50 in ShanghaiTec_partA with input_size 448. How can I fix it?

Yiming-M commented 2 months ago

Hi @BurstLink666 ,

Thanks for report this issue! I forgot to update eval.py, which caused this bug. I have fixed it and things should work fine now! Feel free to reopen this issue if that's not the case.

Many thanks, Yiming