From 7add11150954879dae4daad4e4da549b4b13bca6 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Sat, 09 Aug 2014 15:16:37 +0000 Subject: [PATCH] maxpool fixed, good on mnist --- Makefile | 25 ++++++++++++++----------- 1 files changed, 14 insertions(+), 11 deletions(-) diff --git a/Makefile b/Makefile index 3b01ab2..877fc7f 100644 --- a/Makefile +++ b/Makefile @@ -1,8 +1,12 @@ CC=gcc -GPU=1 -COMMON=-Wall `pkg-config --cflags opencv` -I/usr/local/cuda/include/ +GPU=0 +COMMON=-Wall -Wfatal-errors `pkg-config --cflags opencv` -I/usr/local/cuda/include/ +ifeq ($(GPU), 1) +COMMON+=-DGPU +else +endif UNAME = $(shell uname) -OPTS=-O3 +OPTS=-Ofast -flto ifeq ($(UNAME), Darwin) COMMON+= -isystem /usr/local/Cellar/opencv/2.4.6.1/include/opencv -isystem /usr/local/Cellar/opencv/2.4.6.1/include ifeq ($(GPU), 1) @@ -15,26 +19,25 @@ endif endif CFLAGS= $(COMMON) $(OPTS) -CFLAGS= $(COMMON) -O0 -g +#CFLAGS= $(COMMON) -O0 -g LDFLAGS+=`pkg-config --libs opencv` -lm VPATH=./src/ EXEC=cnn +OBJDIR=./obj/ -OBJ=network.o image.o tests.o connected_layer.o maxpool_layer.o activations.o list.o option_list.o parser.o utils.o data.o matrix.o softmax_layer.o mini_blas.o convolutional_layer.o cpu_gemm.o normalization_layer.o -ifeq ($(GPU), 1) -OBJ+=gpu_gemm.o opencl.o -endif +OBJ=network.o image.o cnn.o connected_layer.o maxpool_layer.o activations.o list.o option_list.o parser.o utils.o data.o matrix.o softmax_layer.o mini_blas.o convolutional_layer.o gemm.o normalization_layer.o opencl.o im2col.o col2im.o axpy.o dropout_layer.o +OBJS = $(addprefix $(OBJDIR), $(OBJ)) all: $(EXEC) -$(EXEC): $(OBJ) +$(EXEC): $(OBJS) $(CC) $(CFLAGS) $(LDFLAGS) $^ -o $@ -%.o: %.c +$(OBJDIR)%.o: %.c $(CC) $(CFLAGS) -c $< -o $@ .PHONY: clean clean: - rm -rf $(OBJ) $(EXEC) + rm -rf $(OBJS) $(EXEC) -- Gitblit v1.10.0