Attila94

name: pytorch

channels:
  - conda-forge
  - pytorch

dependencies:
- python=3.7.7
- pytorch::pytorch=1.7.0
- pytorch::torchvision=0.8.1

Attila94

class MyEarlyStopping(EarlyStopping):

  def __init__(self, minimum_target=None, **kwargs):
    self.minimum_target = minimum_target
    super(MyEarlyStopping, self).__init__(**kwargs)

  def on_epoch_end(self, epoch, logs):
    current = logs.get(self.monitor)
    if self.minimum_target and self.monitor_op(self.minimum_target, current):
          super(MyEarlyStopping, self).on_epoch_end(epoch, logs)

Attila94

# -*- coding: utf-8 -*-
"""
Created on Tue Apr  9 14:33:10 2019

Make 96*96 square crops from bounding boxes of ExDark dataset.

@author: Attila Lengyel
"""

import numpy as np

Attila94

# -*- coding: utf-8 -*-
"""
Created on Thu Apr 11 10:31:43 2019

@author: Attila Lengyel
"""

import h5py

def read_hdf5(path):

Attila94

# -*- coding: utf-8 -*-
"""
Created on Thu Feb 28 19:56:58 2019

@author: Attila Lengyel
attila@lengyel.nl
"""

import numpy as np
import os