Exemplo n.º 1
0
    def init_weights(self):
        # submodule init weights
        self.feature_extractor.init_weights()
        self.rpn_model.init_weights()

        Filler.normal_init(self.rcnn_cls_pred, 0, 0.01, self.truncated)
        Filler.normal_init(self.rcnn_bbox_pred, 0, 0.001, self.truncated)
Exemplo n.º 2
0
    def init_weights(self):
        self.truncated = False

        Filler.normal_init(self.rpn_conv, 0, 0.01, self.truncated)

        for attr_name in self.branches:
            Filler.normal_init(self.branches[attr_name], 0, 0.01,
                               self.truncated)
Exemplo n.º 3
0
    def init_weights(self):
        # submodule init weights
        self.feature_extractor.init_weights()
        self.rpn_model.init_weights()

        # init branches
        for attr_name in self.branches:
            Filler.normal_init(self.branches[attr_name][0], 0, 0.01,
                               self.truncated)
Exemplo n.º 4
0
    def init_weights(self):
        # submodule init weights
        self.feature_extractor.init_weights()
        self.rpn_model.init_weights()

        for stage_ind in range(self.num_stages):
            Filler.normal_init(self.rcnn_cls_preds[stage_ind], 0, 0.01,
                               self.truncated)
            Filler.normal_init(self.rcnn_bbox_preds[stage_ind], 0, 0.001,
                               self.truncated)
Exemplo n.º 5
0
    def init_weights(self):
        self.truncated = False

        Filler.normal_init(self.rpn_conv_cls, 0, 0.01, self.truncated)
        # Filler.normal_init(self.rpn_conv_bbox, 0, 0.01, self.truncated)
        Filler.normal_init(self.rpn_cls_score, 0, 0.01, self.truncated)
        Filler.normal_init(self.rpn_bbox_pred, 0, 0.01, self.truncated)
Exemplo n.º 6
0
    def init_weights(self):
        # submodule init weights
        self.feature_extractor.init_weights()
        self.rpn_model.init_weights()

        Filler.normal_init(self.rcnn_cls_pred, 0, 0.01, self.truncated)
        Filler.normal_init(self.rcnn_bbox_pred, 0, 0.001, self.truncated)

        Filler.normal_init(self.rcnn_coarse_map_conv_iod, 0, 0.001,
                           self.truncated)
        Filler.normal_init(self.rcnn_fine_map_conv_iod, 0, 0.001,
                           self.truncated)

        Filler.normal_init(self.rcnn_coarse_map_conv_iou, 0, 0.001,
                           self.truncated)

        Filler.normal_init(self.rcnn_fine_map_conv_iou, 0, 0.001,
                           self.truncated)
        Filler.normal_init(self.rcnn_fine_map_conv_iog, 0, 0.001,
                           self.truncated)
        Filler.normal_init(self.rcnn_coarse_map_conv_iog, 0, 0.001,
                           self.truncated)

        # freeze all first
        self.freeze_modules()

        # unfreeze some modules
        self.rpn_model.unfreeze_modules()
        self.unfreeze_modules()
Exemplo n.º 7
0
def main():
    """ main entry of this script
    see help option for more features description """
    args = utils.parse_arguments(sys.argv)
    # prepare les logs
    utils.set_logger()

    # option: chargement de la base
    if args.reload:
        Filler().start()
        sys.exit(0)

    if int(args.get_category_by_id) > 0:
        ident = int(args.get_category_by_id)
        dao_category = DaoCategory()
        search_category = dao_category.get_category_by_id(ident)
        if search_category is None:
            print("******************************************************")
            print("Pas de categorie trouvée pour cet identifiant")
            print("******************************************************")
        else:
            print("******************************************************")
            print("Vous avez selectionné la categorie suivante :")
            print("******************************************************")
            print(search_category)
        sys.exit(0)

    if int(args.get_product_by_id) > 0:
        ident = int(args.get_product_by_id)
        dao_product = DaoProduct()
        search_product = dao_product.get_product_by_id(ident)
        if search_product is None:
            print("******************************************************")
            print("Pas de produit trouvé pour cet identifiant")
            print("******************************************************")
        else:
            print("******************************************************")
            print("Vous avez selectionné le produit suivant :")
            print("******************************************************")
            print(search_product)
        sys.exit(0)

    if int(args.get_products_subst_list) > 0:
        ident = int(args.get_products_subst_list)
        dao_product = DaoProduct()
        products = dao_product.get_products_subst_list_by_id(ident)
        search_product = dao_product.get_product_by_id(ident)
        print("******************************************************")
        print("Vous souhaitez un substitut pour le produit suivant :")
        print("******************************************************")
        print(search_product)
        print("******************************************************")
        print("Pur Beurre vous propose le(s) substitut(s) suivant(s) :")
        print("******************************************************")
        for product in products:
            product.list_item()
        sys.exit(0)

    to_match = args.get_products_list_by_match
    if str(to_match) != "":
        dao_product = DaoProduct()
        products = dao_product.get_products_list_by_match(to_match)
        print("******************************************************")
        print("Résultat de la recherche sur les mots clés suivants :")
        print("******************************************************")
        print(to_match)
        print("******************************************************")
        print("Produits correspondants :")
        print("******************************************************")
        for product in products:
            product.list_item()
        sys.exit(0)

    category_id = args.get_product_list_by_category_id
    if str(category_id) != "":
        dao_product = DaoProduct()
        products = dao_product.get_product_list_by_category_id(category_id)
        print("******************************************************")
        print("Résultat de la recherche sur l'id categorie suivant:")
        print("******************************************************")
        print(category_id)
        print("******************************************************")
        print("Produits correspondants :")
        print("******************************************************")
        for product in products:
            product.list_item()
        sys.exit(0)

    if args.get_category_list:
        dao_category = DaoCategory()
        categories = dao_category.get_category_list()
        print("******************************************************")
        print("Résultat de la recherche sur les categories :")
        print("******************************************************")
        for category in categories:
            category.list_item()
        sys.exit(0)

    if args.get_recorded_substitutes_product:
        dao_product = DaoProduct()
        list_substitutes = dao_product.get_recorded_substitutes_product()
        print("******************************************************")
        print(
            "Résultat des substitutions enregistrées (produit => substitut):")
        print("******************************************************")
        for substitute in list_substitutes:
            print('Produit :')
            print(' '.join([
                substitute[ikey] for ikey in substitute.keys()
                if str(ikey).endswith('_PRODUIT')
            ]))
            print('Substitut :')
            print(' '.join([
                substitute[ikey] for ikey in substitute.keys()
                if str(ikey).endswith('_SUBSTITUT')
            ]))
            print('-----------')
        sys.exit(0)

    a_tuple = args.set_substitute_product
    if str(a_tuple) != "":
        r_tuple = tuple(str(a_tuple).split(','))
        Filler().set_substitute_product(r_tuple)
        print("******************************************************")
        print("Insertion de la substitution effectuée.")
        print("******************************************************")
        sys.exit(0)
Exemplo n.º 8
0
    def init_weights(self):
        self.truncated = False

        Filler.normal_init(self.rpn_conv, 0, 0.01, self.truncated)
        Filler.normal_init(self.rpn_cls_score, 0, 0.01, self.truncated)
        Filler.normal_init(self.rpn_bbox_pred, 0, 0.01, self.truncated)

        Filler.normal_init(self.rpn_coarse_map_conv_iod, 0, 0.001,
                           self.truncated)
        Filler.normal_init(self.rpn_fine_map_conv_iod, 0, 0.001,
                           self.truncated)

        Filler.normal_init(self.rpn_coarse_map_conv_iou, 0, 0.001,
                           self.truncated)

        Filler.normal_init(self.rpn_fine_map_conv_iou, 0, 0.001,
                           self.truncated)
        Filler.normal_init(self.rpn_fine_map_conv_iog, 0, 0.001,
                           self.truncated)
        Filler.normal_init(self.rpn_coarse_map_conv_iog, 0, 0.001,
                           self.truncated)
Exemplo n.º 9
0
 def init_weights(self):
     Filler.normal_init(self.toplayer, 0, 0.01, self.truncated)
     Filler.normal_init(self.smooth2, 0, 0.01, self.truncated)
     Filler.normal_init(self.smooth3, 0, 0.01, self.truncated)
     Filler.normal_init(self.smooth4, 0, 0.01, self.truncated)
     Filler.normal_init(self.lateral2, 0, 0.01, self.truncated)
     Filler.normal_init(self.lateral3, 0, 0.01, self.truncated)
     Filler.normal_init(self.lateral4, 0, 0.01, self.truncated)