mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
72 lines
1.7 KiB
C
72 lines
1.7 KiB
C
#include "darknet/network.h"
|
|
#include "darknet/utils.h"
|
|
#include "darknet/parser.h"
|
|
#include "darknet/option_list.h"
|
|
#include "darknet/blas.h"
|
|
#include "darknet/classifier.h"
|
|
#include <sys/time.h>
|
|
|
|
void demo_art(char *cfgfile, char *weightfile, int cam_index)
|
|
{
|
|
#ifdef OPENCV
|
|
network net = parse_network_cfg(cfgfile);
|
|
if(weightfile){
|
|
load_weights(&net, weightfile);
|
|
}
|
|
set_batch_network(&net, 1);
|
|
|
|
srand(2222222);
|
|
CvCapture * cap;
|
|
|
|
cap = cvCaptureFromCAM(cam_index);
|
|
|
|
char *window = "ArtJudgementBot9000!!!";
|
|
if(!cap) error("Couldn't connect to webcam.\n");
|
|
cvNamedWindow(window, CV_WINDOW_NORMAL);
|
|
cvResizeWindow(window, 512, 512);
|
|
int i;
|
|
int idx[] = {37, 401, 434};
|
|
int n = sizeof(idx)/sizeof(idx[0]);
|
|
|
|
while(1){
|
|
image in = get_image_from_stream(cap);
|
|
image in_s = resize_image(in, net.w, net.h);
|
|
show_image(in, window);
|
|
|
|
float *p = network_predict(net, in_s.data);
|
|
|
|
printf("\033[2J");
|
|
printf("\033[1;1H");
|
|
|
|
float score = 0;
|
|
for(i = 0; i < n; ++i){
|
|
float s = p[idx[i]];
|
|
if (s > score) score = s;
|
|
}
|
|
score = score;
|
|
printf("I APPRECIATE THIS ARTWORK: %10.7f%%\n", score*100);
|
|
printf("[");
|
|
int upper = 30;
|
|
for(i = 0; i < upper; ++i){
|
|
printf("%c", ((i+.5) < score*upper) ? 219 : ' ');
|
|
}
|
|
printf("]\n");
|
|
|
|
free_image(in_s);
|
|
free_image(in);
|
|
|
|
cvWaitKey(1);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
void run_art(int argc, char **argv)
|
|
{
|
|
int cam_index = find_int_arg(argc, argv, "-c", 0);
|
|
char *cfg = argv[2];
|
|
char *weights = argv[3];
|
|
demo_art(cfg, weights, cam_index);
|
|
}
|
|
|