nyj-ocean

I have only CPU. Could this code support CPU now?

Is there a script to generate image captions like Figure 5 in the paper

4

Thanks for your great work! I want to know if there is a script to generate image captions like Figure 5 in the paper? 

where is the pretrained models and codes to extract features for new images?

5

Thanks for your great work ! I was interested in your work!  > We also provide pretrained models and codes to extract features for new images. Can you tell...

RuntimeError: CUDA out of memory.

7

## ❓ Questions and Help Thanks for your great work! @KaihuaTang I was very interested in your work! When I train model using the following command > CUDA_VISIBLE_DEVICES=0 python -m...

some attention exist in `yoloair-main/configs/attention/`, such as `yolov5s_TripletAttention.yaml`, `yolov5s_PolarizedSelfAttention.yaml`, `yolov5s_ParNetAttention.yaml`, `yolov5s_GlobalContextBlock.yaml` , etc. however, these attention modules do do not exist in `yoloair-main/models/Models/Attention/` @iscyy