U yÃ9dÔã@s&ddlZddlmZGdd„dƒZdS)éN)Úimage_caption_apic@s2eZdZddœdd„Zdd„Zdd„Zd d „ZdS) ÚDenseCaptioningN)ÚreturncCs d|_dS©N)Úmodel©Úself©r ú5/mnt/home/wangjp/Code/Image2Text/models/grit_model.pyÚ__init__szDenseCaptioning.__init__cCsdSrr rr r r Ú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_captionr r r Úimage_dense_caption_debug sz)DenseCaptioning.image_dense_caption_debugcCs$t|ƒ}tdƒt|ƒtdƒ|S)NzStep2, Dense Caption: ze ****************************************************************************************************)rÚprintr r r r Úimage_dense_captions z#DenseCaptioning.image_dense_caption)Ú__name__Ú __module__Ú __qualname__r r rrr r r r rsr)ÚosZ$models.grit_src.image_dense_captionsrrr r r r Ús