o i~@s`dZddlZddlZddlZddlZddlZddlZddlmZm Z ddl Z ddl m Z ddl mZmZmZmZddlmZddlmZdd lmZdd lmZmZdd lmZmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+dd l,m-Z-e.e/Z0e j1Z2z ddl3Z3e3j4Z2Wn e5yYnwGdddZ6GdddZ7GdddZ8dS)zlProvides an API for writing protocol buffers to event files to be consumed by TensorBoard for visualization.N)OptionalUnion) CometLogger) append_pbtxtmake_mat make_spritemake_tsv)EventFileWriter)load_onnx_graph)load_openvino_graph) event_pb2 summary_pb2)Event SessionLog) audiocustom_scalars histogram histogram_rawhparamsimage image_boxesmeshpr_curve pr_curve_rawscalartextvideo)figure_to_imagec@sveZdZdZddZddZdddZdd d Zdd d Zdd dZ ddZ ddZ ddZ e jdddddZdS)DummyFileWriterz8A fake file writer that writes nothing to the disk. cCs ||_dSN_logdir)selflogdirr%G/home/ubuntu/.local/lib/python3.10/site-packages/tensorboardX/writer.py__init__5s zDummyFileWriter.__init__cCs|jSz7Returns the directory where event file will be written.r!r#r%r%r& get_logdir8szDummyFileWriter.get_logdirNcCdSr r%r#eventstepwalltimer%r%r& add_event<zDummyFileWriter.add_eventcCr+r r%)r#summary global_stepr/r%r%r& add_summary?r1zDummyFileWriter.add_summarycCr+r r%)r# graph_profiler/r%r%r& add_graphBr1zDummyFileWriter.add_graphcCr+r r%)r#graphr/r%r%r&add_onnx_graphEr1zDummyFileWriter.add_onnx_graphcCr+r r%r)r%r%r&flushHr1zDummyFileWriter.flushcCr+r r%r)r%r%r&closeKr1zDummyFileWriter.closecCr+r r%r)r%r%r&reopenNr1zDummyFileWriter.reopenr3r/ccs |VdSr r%r#r3r/r%r%r& use_metadataQs zDummyFileWriter.use_metadataNNr )__name__ __module__ __qualname____doc__r'r*r0r4r6r8r9r:r; contextlibcontextmanagerr>r%r%r%r&r1s    rc@seZdZdZdddZddZdd d Zdd d ZdddZdddZ dddZ ddZ ddZ ddZ ejd d dddZd S) FileWriteraWrites protocol buffers to event files to be consumed by TensorBoard. The `FileWriter` class provides a mechanism to create an event file in a given directory and add summaries and events to it. The class updates the file contents asynchronously. This allows a training program to call methods to add data to the file directly from the training loop, without slowing down training. xcs8t|}t||||_fdd}t|i_dS)adCreates a `FileWriter` and an event file. On construction the writer creates a new event file in `logdir`. The other arguments to the constructor control the asynchronous writes to the event file. Args: logdir: A string. Directory where event file will be written. max_queue: Integer. Size of the queue for pending events and summaries before one of the 'add' calls forces a flush to disk. Default is ten items. flush_secs: Number. How often, in seconds, to flush the pending events and summaries to disk. Default is every two minutes. filename_suffix: A string. Suffix added to all event filenames in the logdir directory. More details on filename construction in tensorboard.summary.writer.event_file_writer.EventFileWriter. csjdSr  event_writerr:r%r)r%r&cleanupysz$FileWriter.__init__..cleanupN)strr rKatexitregister_default_metadata)r#r$ max_queue flush_secsfilename_suffixrLr%r)r&r'`s   zFileWriter.__init__cCs |jSr()rKr*r)r%r%r&r*s zFileWriter.get_logdirNcCsf|dur |jdtn|}|dur||_|dur |jdn|}|dur+t||_|j|dS)aIAdds an event to the event file. Args: event: An `Event` protocol buffer. step: Number. Optional global step value for training process to record with the event. walltime: float. Optional walltime to override the default (current) walltime (from time.time()) Nr/r3)rPgettime wall_timeintr.rKr0r,r%r%r&r0s  zFileWriter.add_eventcCstj|d}||||dS)aAdds a `Summary` protocol buffer to the event file. This method wraps the provided summary in an `Event` protocol buffer and adds it to the event file. Args: summary: A `Summary` protocol buffer. global_step: Number. Optional global step value for training process to record with the summary. walltime: float. Optional walltime to override the default (current) walltime (from time.time()) )r2N)r rr0)r#r2r3r/r-r%r%r&r4s zFileWriter.add_summarycCs^|d}|d}tj|d}||d|tjd|d}tj|d}||d|dS)a&Adds a `Graph` and step stats protocol buffer to the event file. Args: graph_profile: A `Graph` and step stats protocol buffer. walltime: float. Optional walltime to override the default (current) walltime (from time.time()) seconds after epoch rr graph_defNprofiler)tag run_metadata)tagged_run_metadata)r rSerializeToStringr0TaggedRunMetadata)r#r5r/r7 stepstatsr-trmr%r%r&r6s zFileWriter.add_graphcC"tj|d}||d|dSzAdds a `Graph` protocol buffer to the event file. Args: graph: A `Graph` protocol buffer. walltime: float. Optional walltime to override the default (current) _get_file_writerfrom time.time()) rXNr rr^r0r#r7r/r-r%r%r&r8zFileWriter.add_onnx_graphcCrbrcrdrer%r%r&add_openvino_graphrfzFileWriter.add_openvino_graphcC|jdS)zFlushes the event file to disk. Call this method to make sure that all pending events have been written to disk. N)rKr9r)r%r%r&r9szFileWriter.flushcCrh)zFlushes the event file to disk and close the file. Call this method when you do not need the summary writer anymore. NrJr)r%r%r&r:szFileWriter.closecCrh)zReopens the EventFileWriter. Can be called after `close()` to add more events in the same directory. The events will go into a new events file. Does nothing if the EventFileWriter was not closed. N)rKr;r)r%r%r&r;szFileWriter.reopenr<ccsD|jrJd|dus|dusJd||d|_|Vi|_dS)ahContext manager to temporarily set default metadata for all enclosed :meth:`add_event` calls. Args: global_step: Global step value to record walltime: Walltime to record (defaults to time.time()) Examples:: with writer.use_metadata(global_step=10): writer.add_event(event) z Default metadata is already set.Nz=At least one of `global_step` or `walltime` must be provided.r<)rPr=r%r%r&r>s  zFileWriter.use_metadata)rGrHrIr?r )r@rArBrCr'r*r0r4r6r8rgr9r:r;rDrEr>r%r%r%r&rFVs       rFc@seZdZdZddddddddddif deed eed eed eed eed eedeedeedeefddZ ddZ ddZ ddZ  ddeee eeeeffdeeefdeedeefddZ    dded e eefdeed!eed"eed#eef d$d%Z  dd&ed'eeefdeed!eefd(d)Zd*d+Z  ,  dded-edeed.eed!eef d/d0Z  ddedeed!eefd1d2Z   3dded4edeed!eed5eef d6d7Z   8dded4edeed!eed5eef d9d:Z   3 dded4ed;edeed!eed5eedZ   ddedeed?eed!eefd@dAZ  B  CddedDedeedEee eefd!eed5eef dFdGZ  H ddedIedeedJeed!eef dKdLZ  ddedMedeed!eefdNdOZdPdQZdRdSZ  T ddUdVZ e!dWdXZ"    Y ddZed[edeefd\d]Z#  ^  dded