U A=d0ã@s&ddlZddlmZGdd„dƒZdS)éN)Úimage_caption_apic@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚDenseCaptioningcCs ||_dS©N)Údevice)Úselfr©rúQ/mnt/home/wangjp/Code/Image2Text_HuggingFace/Image2Paragraph/models/grit_model.pyÚ__init__szDenseCaptioning.__init__cCsdSrr)rrrrÚinitialize_model sz DenseCaptioning.initialize_modelcCsd}|S)Nz¨ 1. the broccoli is green, [0, 0, 333, 325]; 2. a piece of broccoli, [0, 147, 143, 324]; 3. silver fork on plate, [4, 547, 252, 612]; r©rÚ image_srcÚ dense_captionrrrÚimage_dense_caption_debug sz)DenseCaptioning.image_dense_caption_debugcCs0t||jƒ}tdƒtdƒt|ƒtdƒ|S)Nzo****************************************************************************************************zStep2, Dense Caption: )rrÚprintr rrrÚimage_dense_captions  z#DenseCaptioning.image_dense_captionN)Ú__name__Ú __module__Ú __qualname__r r rrrrrrrsr)ÚosZ$models.grit_src.image_dense_captionsrrrrrrÚs